diff --git a/tests/run.exp b/tests/run.exp
index 87d6347..6991d8a 100755
--- a/tests/run.exp
+++ b/tests/run.exp
@@ -10,36 +10,38 @@ spawn /bin/bash
 # Setup user
 # expect "root@(none):/# "
 # send "mount -o remount,rw /dev/mmcblk0p2 /\r"
-expect "root@(none):/# "
+expect "root@raspberrypi:/# "
 send "passwd pi\r"
 expect "New password:"
 send "$loginPassword\r"
 expect "Retype new password:"
 send "$loginPassword\r"
-expect "root@(none):/# "
-send "sync\r"
-expect "root@(none):/# "
-send "exec /sbin/init\r"
+expect "root@raspberrypi:/# "
+send "sudo su - pi\r"
 
-# Login process
-expect {
-  "raspberrypi login: " {
-    send "$loginUser\r"
-    exp_continue
-  }
-  "Password: " {
-    send "$loginPassword\r"
-    exp_continue
-  }
-  "pi@raspberrypi:~$ " {
-    send "sudo /zram-config/install.bash\r"
-  }
-  "Login incorrect" {
-    exit 1
-  }
-}
+
+# # Login process
+# expect {
+#   "raspberrypi login: " {
+#     send "$loginUser\r"
+#     exp_continue
+#   }
+#   "Password: " {
+#     send "$loginPassword\r"
+#     exp_continue
+#   }
+#   "pi@raspberrypi:~$ " {
+#     send "sudo /zram-config/install.bash\r"
+#   }
+#   "Login incorrect" {
+#     exit 1
+#   }
+# }
 
 # Run tests
+expect "pi@raspberrypi:~$ "
+send "sudo /zram-config/install.bash\r"
+
 expect "pi@raspberrypi:~$ "
 send "/zram-config/test-zram-devices.bash\r"
 expect {