diff --git a/browse.py b/browse.py
index a78dc36f7d5b7cf6c02526d657cc49ccbe34432a..ed91b8681e49c16653bde554903c0cb62ee7a2ea 100755
--- a/browse.py
+++ b/browse.py
@@ -3,6 +3,6 @@ import sys
 import numpy as np
 import matplotlib.pyplot as plt
 
-from KerasROOTClassification import ClassificationProject
+from KerasROOTClassification import *
 
 c = ClassificationProject(sys.argv[1])