diff --git a/src/rest/connector/tests/test_nso.py b/src/rest/connector/tests/test_nso.py
index 61ef477..1c0b3cd 100644
--- a/src/rest/connector/tests/test_nso.py
+++ b/src/rest/connector/tests/test_nso.py
@@ -51,6 +51,8 @@ def test_connect_restconf(self, **kwargs):
 </restconf>
 """
 
+        # Mock /api to return 404 so connect() falls through to /restconf
+        kwargs['mock'].get('http://198.51.100.2:8080/api', status_code=404)
         kwargs['mock'].get('http://198.51.100.2:8080/restconf', text=response_text)
         output = connection.connect(verbose=True).text
         self.assertEqual(output, response_text)