diff --git a/Files/mnist_keras_data_handler.py b/Files/mnist_keras_data_handler.py
index 938f9525..cb97051c 100644
--- a/Files/mnist_keras_data_handler.py
+++ b/Files/mnist_keras_data_handler.py
@@ -46,9 +46,7 @@ def get_data(self, nb_points=500):
                 (x_test, y_test)= pickle.load(f)
                 
             x_train = x_train / 255.0
-            y_train = y_train / 255.0
             x_test = x_test / 255.0
-            y_test = y_test / 255.0
 
             
         except Exception: