path fixed
diff --git a/storage_api.py b/storage_api.py
index 8e995c1..026a46d 100644
--- a/storage_api.py
+++ b/storage_api.py
@@ -2,9 +2,11 @@
 # from gspread import WorksheetNotFound, login
 import json
 import os
-from gspread import login
+from gspread import login, WorksheetNotFound
 from config import ROW_COUNT, DOCUMENT_ID, WORK_SHEET
 
+TEST_PATH = os.environ.get("TEST_PATH", os.path.dirname(__file__) + "/test_results")
+
 
 def get_work_sheet(sheet, name, column_names):
     try: