diff --git a/data_generator.py b/data_generator.py
index 975f307..6cc0678 100644
--- a/data_generator.py
+++ b/data_generator.py
@@ -1,4 +1,5 @@
 import sys
+import os
 import uuid
 import random
 import itertools
@@ -15,7 +16,8 @@
 op_type = ["randread", "read", "randwrite", "write"]
 is_sync = ["s", "a"]
 
-storage = create_storage("file:///home/gstepanov/rally-results-processor/sample.json", "", "")
+
+storage = create_storage("file://" + os.path.dirname(__file__) + "/sample.json", "", "")
 combinations = list(itertools.product(op_type, is_sync, sz))
 
 for i in range(30):
@@ -27,5 +29,5 @@
         row[" ".join([sz, op_type, is_sync])] = (random.random() * 100,
                                                  random.random() * 5)
 
-    print row
+    print len(row)
     storage.store(row)