diff --git a/examples/bert/bert_classifier.py b/examples/bert/bert_classifier.py
index 6bb9ea3beec92d5de6af703ac8b2f2a316e7f9ab..ef43ae2076e665a88fd896dd7e6f830c9e38640c 100644
--- a/examples/bert/bert_classifier.py
+++ b/examples/bert/bert_classifier.py
@@ -103,7 +103,7 @@ def main():
         batch_size=config.batch_size,
         line_processor=mnli_line_processor)
 
-    dev_dataloader = BertDataLoader(
+    test_dataloader = BertDataLoader(
         "./data/glue_data/MNLI/dev_matched.tsv",
         tokenizer, ["contradiction", "entailment", "neutral"],
         max_seq_length=config.max_seq_len,
diff --git a/examples/bert_leveldb/bert_classifier.py b/examples/bert_leveldb/bert_classifier.py
index 891901388d742623efc9a75f62fa11f850210856..11bc85758ebbe81ae68b3c141d4582ee8d41508c 100644
--- a/examples/bert_leveldb/bert_classifier.py
+++ b/examples/bert_leveldb/bert_classifier.py
@@ -105,7 +105,7 @@ def main():
         mode="leveldb",
         phase="train")
 
-    dev_dataloader = BertDataLoader(
+    test_dataloader = BertDataLoader(
         "./data/glue_data/MNLI/dev_matched.tsv",
         tokenizer, ["contradiction", "entailment", "neutral"],
         max_seq_length=config.max_seq_len,
diff --git a/hapi/text/bert/optimization.py b/hapi/text/bert/optimization.py
index 2bf6b7f2621273ff78bc88a6ef92f1f630072175..b2ba8f65a744754e8ff96ca66ccf818bc8b06c34 100755
--- a/hapi/text/bert/optimization.py
+++ b/hapi/text/bert/optimization.py
@@ -130,6 +130,18 @@ class Optimizer(object):
                 return True
         return False
 
+    def state_dict(self):
+        return self.optimizer.state_dict()
+
+    def set_dict(self, state_dict):
+        return self.optimizer.set_dict(state_dict)
+
+    def get_opti_var_name_list(self):
+        return self.optimizer.get_opti_var_name_list()
+
+    def current_step_lr(self):
+        return self.optimizer.current_step_lr()
+
     def minimize(self, loss, use_data_parallel=False, model=None):
         param_list = dict()