diff --git a/scripts/post_worker.py b/scripts/post_worker.py
index d4994b382d..7cba7640d4 100644
--- a/scripts/post_worker.py
+++ b/scripts/post_worker.py
@@ -27,4 +27,10 @@ if count > 0:
     shutil.copyfile(output_filename, file_dest)
     print "done copying file"
 
+# TODO: fold the remaining bash script into the Python script
+# finding if there is any screenshots
+# print "checking for screenshots"
+# for (path, dirs, files) in os.walk('test_root/log'):
+#    print files
+
 print "TDDIUM_SESSION_ID:", os.environ['TDDIUM_SESSION_ID']
\ No newline at end of file
diff --git a/tddium.yml b/tddium.yml
index 98a33737af..c0d2a22527 100644
--- a/tddium.yml
+++ b/tddium.yml
@@ -7,9 +7,8 @@ tddium:
       :version: "1.8.5.5"
   :hooks:
     :pre_setup: "virtualenv $HOME/python-env && $HOME/python-env/bin/pip install -r requirements/edx/paver.txt && $HOME/python-env/bin/pip install -r requirements/edx/pre.txt && $HOME/python-env/bin/pip install -r requirements/edx/base.txt && $HOME/python-env/bin/pip install -r requirements/edx/github.txt && $HOME/python-env/bin/pip install -r requirements/edx/local.txt && $HOME/python-env/bin/pip install -r requirements/edx/post.txt"
-    :post_build: 'for i in $(find test_root/log -name *.png); do cp -R $i $HOME/results/$TDDIUM_SESSION_ID/session/; echo "copying file $i"; done'
     # :post_build: "paver coverage; paver bokchoy_coverage; for i in $(find reports -name cover); do cp -R $i $HOME/results/$TDDIUM_SESSION_ID/session/; done"
-    :post_worker: 'python ./scripts/post_worker.py'
+    :post_worker: 'python ./scripts/post_worker.py; echo "attaching failed screenshots (if any)"; for i in $(find test_root/log -name *.png); do cp -R $i $HOME/results/$TDDIUM_SESSION_ID/session/; echo "copying file $i"; done'
   :cache:
     :key_paths:
       - requirements/edx/paver.txt