diff --git a/maas-agent-charm/tests/unit/test_charm.py b/maas-agent-charm/tests/unit/test_charm.py
index 438fa22..db9409b 100644
--- a/maas-agent-charm/tests/unit/test_charm.py
+++ b/maas-agent-charm/tests/unit/test_charm.py
@@ -36,6 +36,7 @@ def test_start(self, mock_helper):
 
     @patch("charm.MaasHelper", autospec=True)
     def test_remove(self, mock_helper):
+        mock_helper.get_maas_mode.return_value = "rack"
         self.harness.begin()
         self.harness.charm.on.remove.emit()
         mock_helper.uninstall.assert_called_once()