data gen fixed
diff --git a/data_generator.py b/data_generator.py
index 6cc0678..ff6d927 100644
--- a/data_generator.py
+++ b/data_generator.py
@@ -5,7 +5,7 @@
import itertools
from petname import Generate as pet_generate
-from storage_api import create_storage
+from storage_api import create_storage, TEST_PATH
types = ["GA", "master"] + [pet_generate(2, '-') for _ in range(2)]
@@ -17,7 +17,7 @@
is_sync = ["s", "a"]
-storage = create_storage("file://" + os.path.dirname(__file__) + "/sample.json", "", "")
+storage = create_storage("file://" + TEST_PATH + "/sample.json", "", "")
combinations = list(itertools.product(op_type, is_sync, sz))
for i in range(30):