diff --git a/mothership/internal/api/security_test.go b/mothership/internal/api/security_test.go
index b243289..3c99c2e 100644
--- a/mothership/internal/api/security_test.go
+++ b/mothership/internal/api/security_test.go
@@ -23,6 +23,7 @@ type mockDetectorProvider struct {
 	activeAnomalies  []*events.AnomalyEvent
 	history          []*events.AnomalyEvent
 	modeChanges      []analytics.SecurityMode
+	systemMode       events.SystemMode
 }
 
 func (m *mockDetectorProvider) GetSecurityMode() analytics.SecurityMode {
@@ -67,6 +68,10 @@ func (m *mockDetectorProvider) CountAnomaliesSince(since time.Time) (int, error)
 	return count, nil
 }
 
+func (m *mockDetectorProvider) GetSystemMode() events.SystemMode {
+	return m.systemMode
+}
+
 func TestSecurityHandler_Status(t *testing.T) {
 	tests := []struct {
 		name           string