diff --git a/zaza/charm_tests/dragent/tests.py b/zaza/charm_tests/dragent/tests.py
index 23244a9..3d993c7 100644
--- a/zaza/charm_tests/dragent/tests.py
+++ b/zaza/charm_tests/dragent/tests.py
@@ -2,7 +2,7 @@
 
 import unittest
 
-from zaza.utilities import generic as generic_utils
+from zaza.utilities import cli as cli_utils
 from zaza.charm_tests.dragent import test
 
 
@@ -12,7 +12,7 @@ class DRAgentTest(unittest.TestCase):
 
     @classmethod
     def setUpClass(cls):
-        generic_utils.setup_logging()
+        cli_utils.setup_logging()
 
     def test_bgp_routes(self):
         test.test_bgp_routes(peer_application_name=self.BGP_PEER_APPLICATION)