diff --git a/pysegcnn/core/graphics.py b/pysegcnn/core/graphics.py
index b9fd6e2e173922ed29dd63d3bdddb3655af73664..de88ee0d9af2598570d3e010365ea2b133016f11 100644
--- a/pysegcnn/core/graphics.py
+++ b/pysegcnn/core/graphics.py
@@ -18,7 +18,6 @@ License
 import os
 import pathlib
 import itertools
-import datetime
 
 # externals
 import numpy as np
@@ -33,7 +32,7 @@ from matplotlib import cm as colormap
 
 # locals
 from pysegcnn.core.utils import accuracy_function, check_filename_length
-from pysegcnn.main.config import HERE
+from pysegcnn.main.train_config import HERE
 
 # plot font size configuration
 SMALL = 10