diff --git a/test/integration.nix b/test/integration.nix
index 7cb840c..90f8b8b 100644
--- a/test/integration.nix
+++ b/test/integration.nix
@@ -45,9 +45,9 @@ import "${nixpkgs}/nixos/tests/make-test-python.nix" ({ pkgs, ...}: {
     system1.wait_until_succeeds("[ $(fgconsole) = 2 ]")
     system1.wait_for_unit("getty@tty2.service")
     system1.wait_until_succeeds("pgrep -f 'agetty.*tty2'")
-    system1.wait_until_tty_matches(2, "login: ")
+    system1.wait_until_tty_matches("2", "login: ")
     system1.send_chars("${user}\n")
-    system1.wait_until_tty_matches(2, "login: ${user}")
+    system1.wait_until_tty_matches("2", "login: ${user}")
     system1.wait_until_succeeds("pgrep login")
     system1.sleep(2)
     system1.send_chars("${password}\n")