diff --git a/pywatts/test_kcross_train.py b/pywatts/test_kcross_train.py
index 6201fa2..4e29d0a 100644
--- a/pywatts/test_kcross_train.py
+++ b/pywatts/test_kcross_train.py
@@ -35,7 +35,7 @@ if TRAIN:
 
 if PLOT:
     # Plot training success rate (with 'average loss')
-    pywatts.main.plot_training(train_eval)
+    pywatts.routines.plot_training(train_eval)
 
 
 exit()