diff --git a/test/system/datasettest.py b/test/system/datasettest.py
index 4d2032a..2609c95 100644
--- a/test/system/datasettest.py
+++ b/test/system/datasettest.py
@@ -1,7 +1,7 @@
 import unittest
 import sys
-sys.path.append('../../system')
-from dataset import *
+sys.path.append('../..')
+from system.dataset import *
 from scipy import rand,shape
 
 class DatasetTest(unittest.TestCase):