diff --git a/toolkit.py b/toolkit.py
index d708ee11ac90375624ffc3b7fb0218562d73d677..ee64a09a261aa66a5f173ec9420f54f859ec9129 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -490,9 +490,9 @@ class KerasROOTClassification(object):
         np.random.shuffle(self.y_train)
         np.random.set_state(rn_state)
         np.random.shuffle(self.w_train)
-        if self._scores_test is not None:
+        if self._scores_train is not None:
             np.random.set_state(rn_state)
-            np.random.shuffle(self._scores_test)
+            np.random.shuffle(self._scores_train)
 
 
     def train(self, epochs=10):