diff --git a/unit_tests/test_zaza_model.py b/unit_tests/test_zaza_model.py
index 0b8e37f..95bd7cb 100644
--- a/unit_tests/test_zaza_model.py
+++ b/unit_tests/test_zaza_model.py
@@ -134,7 +134,7 @@ class TestModel(ut_utils.BaseTestCase):
         self.get_current_model.return_value = 'modelsmodel'
 
         def _get_env(key):
-            return _env.get(key)
+            return _env[key]
         self.environ.__getitem__.side_effect = _get_env
         _env = {"JUJU_MODEL": 'envmodel'}
 
@@ -148,7 +148,7 @@ class TestModel(ut_utils.BaseTestCase):
         self.get_current_model.return_value = 'modelsmodel'
 
         def _get_env(key):
-            return _env.get(key)
+            return _env[key]
         self.environ.__getitem__.side_effect = _get_env
         _env = {"MODEL_NAME": 'envmodel'}