diff --git a/deepray/core/base_trainer.py b/deepray/core/base_trainer.py
index bd2f75d..aca2bcf 100644
--- a/deepray/core/base_trainer.py
+++ b/deepray/core/base_trainer.py
@@ -258,7 +258,7 @@ def __init__(
 
     # TODO: fuhailin 
     # if isinstance(optimizer, optimizers.Optimizer):
-      self.optimizer = optimizer
+    self.optimizer = optimizer
     # else:
     #   raise ValueError("Not support opt.")
     self.use_float16 = common_flags.use_float16()
diff --git a/deepray/optimizers/__init__.py b/deepray/optimizers/__init__.py
index f9b81f1..8eaff46 100644
--- a/deepray/optimizers/__init__.py
+++ b/deepray/optimizers/__init__.py
@@ -49,4 +49,4 @@
 )
 from deepray.optimizers.yogi import Yogi
 from deepray.optimizers.cocob import COCOB
-from deepray.optimizers.adam import AdamOptimizer
+from deepray.optimizers.adam import Adam