diff --git a/src/trainer.py b/src/trainer.py
index 205dfec..97a0d2e 100644
--- a/src/trainer.py
+++ b/src/trainer.py
@@ -118,10 +118,10 @@ def test(emoji_data, sess):
     if not os.path.exists(CHECKPOINT_SAVE_PATH):
         os.makedirs(CHECKPOINT_SAVE_PATH)
 
-    BATCH_SIZE = config_parser.getInt('MODEL_HYPER_PARAMETERS', 'batch_size')
-    STEPS = config_parser.getInt('MODEL_HYPER_PARAMETERS', 'train_steps')
-    LEARNING_RATE = config_parser.getFloat('MODEL_HYPER_PARAMETERS', 'learning_rate')
-    KEEP_PROB = config_parser.getFloat('MODEL_HYPER_PARAMETERS', 'dropout_keep_prob')
+    BATCH_SIZE = config_parser.getint('MODEL_HYPER_PARAMETERS', 'batch_size')
+    STEPS = config_parser.getint('MODEL_HYPER_PARAMETERS', 'train_steps')
+    LEARNING_RATE = config_parser.getfloat('MODEL_HYPER_PARAMETERS', 'learning_rate')
+    KEEP_PROB = config_parser.getfloat('MODEL_HYPER_PARAMETERS', 'dropout_keep_prob')
 
     X = tf.placeholder(
         tf.float32, shape=[None, 48, 48, 1]