diff --git a/tests/recovery-raw-disk/recovery_raw_disk_test.go b/tests/recovery-raw-disk/recovery_raw_disk_test.go
index 2cbff862269..68afd210a08 100644
--- a/tests/recovery-raw-disk/recovery_raw_disk_test.go
+++ b/tests/recovery-raw-disk/recovery_raw_disk_test.go
@@ -26,7 +26,7 @@ var _ = Describe("cOS Recovery deploy tests", func() {
 			err = s.ChangeBoot(sut.Active)
 			Expect(err).ToNot(HaveOccurred())
 
-			s.Reboot()
+			s.Reboot(sut.TimeoutRawDiskTest)
 			ExpectWithOffset(1, s.BootFrom()).To(Equal(sut.Active))
 		})
 	})
diff --git a/tests/sut/sut.go b/tests/sut/sut.go
index 32ccbcb629f..3d55267ad0f 100644
--- a/tests/sut/sut.go
+++ b/tests/sut/sut.go
@@ -187,10 +187,10 @@ func (s *SUT) command(cmd string, timeout bool) (string, error) {
 }
 
 // Reboot reboots the system under test
-func (s *SUT) Reboot() {
+func (s *SUT) Reboot(t ...int) {
 	s.command("reboot", true)
 	time.Sleep(10 * time.Second)
-	s.EventuallyConnects(180)
+	s.EventuallyConnects(t...)
 }
 
 func (s *SUT) clientConfig() *ssh.ClientConfig {