diff --git a/whitebox_neutron_tempest_plugin/tests/scenario/test_internal_dns.py b/whitebox_neutron_tempest_plugin/tests/scenario/test_internal_dns.py
index 37adcf7..fbb6bf0 100644
--- a/whitebox_neutron_tempest_plugin/tests/scenario/test_internal_dns.py
+++ b/whitebox_neutron_tempest_plugin/tests/scenario/test_internal_dns.py
@@ -31,7 +31,7 @@ WB_CONF = CONF.whitebox_neutron_plugin_options
 LOG = log.getLogger(__name__)
 
 
-class InternalDNSBaseCommon(base.BaseTempestWhiteboxTestCase):
+class InternalDNSBaseCommon(base.TrafficFlowTest):
     """Common base class of resources and functionalities for test classes."""
 
     port_error_msg = ('Openstack command returned incorrect'