diff --git a/compare.py b/compare.py
index f0a2a3db882b8e90a9edc782e7fc7a3330871cb9..e6496db3fc2e72170d2979729572e160cf7e0c3c 100755
--- a/compare.py
+++ b/compare.py
@@ -7,7 +7,7 @@ import numpy as np
 import matplotlib.pyplot as plt
 from sklearn.metrics import roc_curve, auc
 
-from toolkit import ClassificationProject
+from .toolkit import ClassificationProject
 
 """
 A few functions to compare different setups
diff --git a/toolkit.py b/toolkit.py
index 271989d9e97e334cf56ca8a12be0b12705286988..a0d3a1c04c798f6722385772088f4684f74bfb81 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -764,6 +764,8 @@ if __name__ == "__main__":
 
     filename = "/project/etp4/nhartmann/trees/allTrees_m1.8_NoSys.root"
 
+
+
     c = ClassificationProject("test4",
                               signal_trees = [(filename, "GG_oneStep_1705_1105_505_NoSys")],
                               bkg_trees = [(filename, "ttbar_NoSys"),