diff --git a/docker/scripts/run-ansible-playbook.py b/docker/scripts/run-ansible-playbook.py
index abd8cd1..c33439b 100644
--- a/docker/scripts/run-ansible-playbook.py
+++ b/docker/scripts/run-ansible-playbook.py
@@ -59,7 +59,7 @@ def execute_playbook():
     
     rc = RunnerConfig(
         private_data_dir="/data",
-        project_dir="/workspace"
+        project_dir=os.environ.get('ANSIBLE_PLAYBOOK_WORKSPACE_PATH', "/workspace")
     )
     
     rc.playbook=os.environ.get('ANSIBLE_PLAYBOOK', "site.yml")
@@ -77,10 +77,10 @@ def execute_playbook():
     r = Runner(config=rc)
     r.run()
 
-def list_workspace():
+def list_workspace(rc):
     print("---------------------------------------")
     print("workspace files:")
-    items=os.listdir("/workspace")
+    items=os.listdir(rc.project_dir)
     for image in items:
         print(image)
     print("---------------------------------------")