diff --git a/test/integration/smoke/test_affinity_groups.py b/test/integration/smoke/test_affinity_groups.py
index e0e1a17273b..e63b91672c6 100644
--- a/test/integration/smoke/test_affinity_groups.py
+++ b/test/integration/smoke/test_affinity_groups.py
@@ -189,6 +189,6 @@ class TestDeployVmWithAffinityGroup(cloudstackTestCase):
             try:
                 #cls.api_client = super(TestDeployVmWithAffinityGroup, cls).getClsTestClient().getApiClient()
                 #Clean up, terminate the created templates
-                cleanup_resources(cls.api_client, cls.cleanup)
+                cleanup_resources(cls.api_client, cls._cleanup)
             except Exception as e:
                 raise Exception("Warning: Exception during cleanup : %s" % e)