diff --git a/test/integration/component/test_persistent_networks.py b/test/integration/component/test_persistent_networks.py
index 69ed493fb15..40703789693 100644
--- a/test/integration/component/test_persistent_networks.py
+++ b/test/integration/component/test_persistent_networks.py
@@ -2780,7 +2780,8 @@ class TestVPCNetworkOperations(cloudstackTestCase):
             # Check if source nat IP address is released
             ipaddresses = PublicIPAddress.list(
                 self.apiclient,
-                id=publicipaddresses[0].id)
+                id=publicipaddresses[0].id,
+                state="Allocated")
             self.assertEqual(
                 validateList(ipaddresses)[0],
                 FAIL,