diff --git a/DeepDrug.py b/DeepDrug.py
index db9688f..aa85e5e 100644
--- a/DeepDrug.py
+++ b/DeepDrug.py
@@ -239,7 +239,7 @@ tf.test.is_gpu_available()
 # In[ ]:
 
 
-history_mild_2mp = mild_model.fit(X_train, Y_train, validation_data=(X_test, Y_test), epochs=30, batch_size=32)
+history_mild_2mp = my_model.fit(X_train, Y_train, validation_data=(X_test, Y_test), epochs=30, batch_size=32)
 my_model.save('new_model_e30_b32_t1000.h5')