1 files changed, 5 insertions, 3 deletions
diff --git a/commands/test-ui b/commands/test-ui
index fb1233e..57a4d35 100755
--- a/commands/test-ui
+++ b/commands/test-ui
@@ -1,13 +1,15 @@
 #!/usr/bin/python
 
 import os
+import sys
 
 import common
 
 from devbot import run
-from devbot import config
 
 common.setup()
 
-test = os.path.join(common.tests_dir, "sugar", "shell.py")
-run.run(["sr-run-test", test, config.logs_dir])
+virtual = "SUGAR_BUILDBOT" in os.environ
+test_path = os.path.join(common.tests_dir, "sugar", "shell.py")
+if not run.run_test(test_path, virtual):
+    sys.exit(1)