diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index e20b7527665c6d5c4b6539a23a19620148ae7cf3..818f2f49607d231a27d9c0efdcfad2c6054958af 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -673,7 +673,7 @@ class IcingIntensityNN:
                 pickle.dump(self.in_mem_data_cache, f)
                 f.close()
 
-            if es.check_stop(self.test_loss.result().numpy()):
+            if EARLY_STOP and es.check_stop(self.test_loss.result().numpy()):
                 break
 
         print('total time: ', total_time)