diff --git a/demos/tgen/test_tgen_demo.py b/demos/tgen/test_tgen_demo.py
index 354d2882..1a29df75 100644
--- a/demos/tgen/test_tgen_demo.py
+++ b/demos/tgen/test_tgen_demo.py
@@ -1,3 +1,4 @@
+import os
 import grpc
 import netmiko
 import pydpu
@@ -7,8 +8,9 @@
 
 from testbed import *
 
-HOME_FOLDER = r'/home/opi/actions-runner/_work/opi-poc/opi-poc'
+#HOME_FOLDER = r'/home/opi/actions-runner/_work/opi-poc/opi-poc'
 #HOME_FOLDER = r'/home/opi/opi-poc'
+HOME_FOLDER = os.environ['GITHUB_WORKSPACE'] 
 
 
 @pytest.fixture