diff --git a/test/test_ims_data.py b/test/test_ims_data.py
index fdf590e23fe75a71523d852ee210985dde12cc0a..a84590d621bcf6272da510c979cc8634aead6c8a 100644
--- a/test/test_ims_data.py
+++ b/test/test_ims_data.py
@@ -96,6 +96,7 @@ def test_get_port_details(mocker):
     ds = inventory_provider.db.ims.IMS(
         'dummy_base', 'dummy_username', 'dummy_password')
     res = list(get_port_details(ds))
+    assert ds.get_all_entities.call_count == 2
     assert _external.call_count == 1
     assert _internal.call_count == 1