diff --git a/files/usr/bin/test-upgrade b/files/usr/bin/test-upgrade
index 24b42e69..28a5040f 100755
--- a/files/usr/bin/test-upgrade
+++ b/files/usr/bin/test-upgrade
@@ -1,2 +1,4 @@
 #!/bin/bash
-cos-upgrade --no-verify --docker-image ibuildthecloud/test
+set -e -x
+sudo cos-upgrade --no-verify --docker-image ibuildthecloud/test
+sudo reboot
diff --git a/files/usr/bin/yip b/files/usr/bin/yip
new file mode 100755
index 00000000..a188e02d
Binary files /dev/null and b/files/usr/bin/yip differ