--- a/testing/docker/tester-device/bin/entrypoint
+++ b/testing/docker/tester-device/bin/entrypoint
@@ -18,9 +18,19 @@ export ADB_HOST=`echo $res | jq .proxies
 export ADB_PORT=`echo $res | jq .proxies.adb.port`
 export MARIONETTE_HOST=`echo $res | jq .proxies.marionette.forwardHost`
 export MARIONETTE_PORT=`echo $res | jq .proxies.marionette.port`
 export PROXY_HOST=`echo $res | jq -r .proxyHost`
 echo "Retrieved device.  Session: $SESSION_ID"
 
 curl -o /home/worker/data/device.json -s -H "Accept: application/json" http://$CLOUD_HOST/device/properties
 
-eval $@
+buildbot_step 'Gaia clone' tc-vcs checkout /home/worker/gaia/source $GAIA_BASE_REPOSITORY $GAIA_HEAD_REPOSITORY $GAIA_REV $GAIA_REF
+
+cd gaia/source/tests/python/gaia-ui-tests/
+python setup.py develop
+pip install py
+cd /home/worker/
+
+buildbot_step 'Running tests' eval $@
+
+echo "Releasing device."
+curl --request POST http://$CLOUD_HOST/device/release

--- a/testing/taskcluster/tasks/tests/flame_kk_gaia_ui_test_sanity.yml
+++ b/testing/taskcluster/tasks/tests/flame_kk_gaia_ui_test_sanity.yml
@@ -10,23 +10,17 @@ task:
 
   payload:
     env:
       DEVICE_CAPABILITIES: '{"type":"flame","memory":"319","sims": "1","build":"{{{img_url}}}"}'
     features:
       testdroidProxy: true
     maxRunTime: 7200
     command:
-      - >
-        tc-vcs checkout /home/worker/gaia/source $GAIA_BASE_REPOSITORY $GAIA_HEAD_REPOSITORY $GAIA_REV $GAIA_REF &&
-        cd gaia/source/tests/python/gaia-ui-tests/ &&
-        python setup.py develop &&
-        pip install py &&
-        cd /home/worker/ &&
-        gaiatest --testvars=/home/worker/data/gaia_testvars.json --testvars=/home/worker/data/device.json --adb-host=$PROXY_HOST --adb-port=$ADB_PORT --address=$PROXY_HOST:$MARIONETTE_PORT --device $SERIAL_ID --xml-output=/home/worker/upload/logs/xml_output.xml --timeout=10000 --log-html=/home/worker/upload/logs/index.html --restart --type=b2g+sanity-dsds --log-mach=-  --log-raw=/home/worker/upload/logs/raw.log gaia/source/tests/python/gaia-ui-tests/gaiatest/tests/functional/manifest.ini
+      - gaiatest --testvars=/home/worker/data/gaia_testvars.json --testvars=/home/worker/data/device.json --adb-host=$PROXY_HOST --adb-port=$ADB_PORT --address=$PROXY_HOST:$MARIONETTE_PORT --device $SERIAL_ID --xml-output=/home/worker/upload/logs/xml_output.xml --timeout=10000 --log-html=/home/worker/upload/logs/index.html --restart --type=b2g+sanity-dsds --log-mach=-  --log-raw=/home/worker/upload/logs/raw.log gaia/source/tests/python/gaia-ui-tests/gaiatest/tests/functional/manifest.ini
     artifacts:
       'private/device.json':
         type: file
         path: '/home/worker/data/device.json'
         expires: '{{#from_now}}1 year{{/from_now}}'
 
   extra:
     treeherder: