diff --git a/charms_openstack/test_mocks.py b/charms_openstack/test_mocks.py
index c25dc15..c20da4a 100644
--- a/charms_openstack/test_mocks.py
+++ b/charms_openstack/test_mocks.py
@@ -76,4 +76,5 @@ def mock_charmhelpers():
         'train',
         'ussuri',
         'victoria',
+        'wallaby',
     )