diff --git a/test/TestGraphBuilder.py b/test/TestGraphBuilder.py
index cbc151f..7c345f7 100644
--- a/test/TestGraphBuilder.py
+++ b/test/TestGraphBuilder.py
@@ -1,3 +1,6 @@
+import sys, os
+sys.path.append(os.path.join(os.path.dirname(sys.path[0]), '../graph_builder', 'graph'))
+
 import unittest
 from graph.GraphBuilder import GraphBuilder