diff --git a/tests/test_diana_utils.py b/tests/test_diana_utils.py
index 882e4ea..3508f90 100644
--- a/tests/test_diana_utils.py
+++ b/tests/test_diana_utils.py
@@ -212,7 +212,7 @@ def test_generate_hana_config(self, confirm, prompt):
         self.assertEqual(hana_config['auth_requests_per_minute'], 6)
 
         # Test none confirmed
-        confirm.return_value = False
+        confirm.side_effect = [False, True]
         prompt.side_effect = [30, 3]
         hana_config = generate_hana_config()
         self.assertIsInstance(hana_config['hana'], dict)