diff --git a/libra/modeling/prediction_model_creation.py b/libra/modeling/prediction_model_creation.py
index 8e24cd40..86ba7f36 100644
--- a/libra/modeling/prediction_model_creation.py
+++ b/libra/modeling/prediction_model_creation.py
@@ -33,7 +33,7 @@ def get_keras_model_reg(dataset, i):
         else:
             model.add(
                 Dense(
-                    i * 64,
+                    64,
                     kernel_initializer="normal",
                     activation="relu"))
 
@@ -68,7 +68,7 @@ def get_keras_model_class(dataset, i, num_classes):
         else:
             model.add(
                 Dense(
-                    i * 64,
+                    64,
                     kernel_initializer="normal",
                     activation="relu"))