diff --git a/toolkit.py b/toolkit.py
index 9955e0d06c77b3bbf91d8756401c22453c729211..f679f69b1e7e56a2bf0eac46cda2649f44006f9a 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -263,8 +263,6 @@ class KerasROOTClassification:
 
         if not self.data_loaded:
             self._load_data()
-        
-        self.scaler
 
         if not self.data_transformed:
             self._transform_data()
@@ -289,14 +287,14 @@ class KerasROOTClassification:
                        class_weight=self.class_weight,
                        shuffle=True,
                        batch_size=self.batch_size)
-        
+
         logger.info("Save weights")
         self.model.save_weights(os.path.join(self.project_dir, "weights.h5"))
 
         self.total_epochs += epochs
         self._write_info("epochs", self.total_epochs)
-       
-        logger.info("Create scores for ROC curve") 
+
+        logger.info("Create scores for ROC curve")
         self.scores_test = self.model.predict(self.x_test)
         self.scores_train = self.model.predict(self.x_train)