diff --git a/pkg/handlers/apitests.go b/pkg/handlers/apitests.go
index 6ab3bbd8679..8250facc90e 100644
--- a/pkg/handlers/apitests.go
+++ b/pkg/handlers/apitests.go
@@ -106,7 +106,13 @@ func (suite *BaseHandlerTestSuite) HandlerConfig() *Config {
 		mock.AnythingOfType("string"),
 		mock.Anything,
 	).Return(func(ctx context.Context, appCtx appcontext.AppContext, key string, flagContext map[string]string) (services.FeatureFlag, error) {
-		return mockGetFlagFunc(ctx, appCtx.Logger(), "user@example.com", key, flagContext, "")
+		return services.FeatureFlag{
+			Entity:    "user@example.com",
+			Key:       key,
+			Match:     true,
+			Variant:   "",
+			Namespace: "test",
+		}, nil
 	})
 	return &Config{
 		db:                 suite.DB(),