diff --git a/tests/lib/then_steps.py b/tests/lib/then_steps.py
index 5502787d..31d5631f 100644
--- a/tests/lib/then_steps.py
+++ b/tests/lib/then_steps.py
@@ -190,12 +190,12 @@ def config_var_in_memory(
 
 @then("we should be prompted for a password")
 def password_was_called(cli_run):
-    assert cli_run["mocks"]["user_input"].called
+    assert cli_run["mocks"]["user_input"].return_value.input.called
 
 
 @then("we should not be prompted for a password")
 def password_was_not_called(cli_run):
-    assert not cli_run["mocks"]["user_input"].called
+    assert not cli_run["mocks"]["user_input"].return_value.input.called
 
 
 @then(parse("the cache directory should contain the files\n{file_list}"))