From 525298f76114990fec3b395de0311ad4976f3a38 Mon Sep 17 00:00:00 2001
From: Paul Schaub <vanitasvitae@fsfe.org>
Date: Mon, 13 Aug 2018 17:17:37 +0200
Subject: [PATCH] Fix reference to old main class

---
 pywatts/test_kcross_train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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()