diff --git a/zaza/openstack/charm_tests/mysql/test_prometheus_mysql_exporter.py b/zaza/openstack/charm_tests/mysql/test_prometheus_mysql_exporter.py
index 723bea9..1ff5e37 100644
--- a/zaza/openstack/charm_tests/mysql/test_prometheus_mysql_exporter.py
+++ b/zaza/openstack/charm_tests/mysql/test_prometheus_mysql_exporter.py
@@ -118,7 +118,7 @@ class PrometheusMySQLExporterTest(MySQLBaseTest):
             self.assertEqual(
                 json_mysql_config,
                 {
-                    "host": unit.public_address,
+                    "host": zaza_model.get_unit_public_address(unit),
                     "port": 3306,
                     "user": "prom_exporter"
                 }