diff --git a/test_runlike.py b/test_runlike.py
index 29d2a46..a4e45f8 100644
--- a/test_runlike.py
+++ b/test_runlike.py
@@ -84,7 +84,7 @@ def test_restart_with_max(self):
     def test_restart_not_present(self):
         # If the restart policy is not set, the default value is no.
         # self.dont_expect_substr('--restart', 4)
-        self.expect_substr('--restart=no \\')
+        self.expect_substr('--restart=no \\', 4)
 
     def test_hostname(self):
         self.expect_substr('--hostname=Essos \\')