diff --git a/model_search/metadata/ml_metadata_db.py b/model_search/metadata/ml_metadata_db.py
index c5636911..6e9a6651 100644
--- a/model_search/metadata/ml_metadata_db.py
+++ b/model_search/metadata/ml_metadata_db.py
@@ -15,6 +15,7 @@
 
 import json
 import os
+import sys
 import random
 from absl import flags
 from absl import logging
@@ -50,7 +51,7 @@
     "config with this password is established.")
 
 FLAGS = flags.FLAGS
-
+FLAGS(sys.argv)
 
 class MLMetaData(metadata.MetaData):
   """An object which handles communicating with metadata db through MLMD."""