diff --git a/pkg/sdk/reports_test.go b/pkg/sdk/reports_test.go
index cfeaee760..fb210a375 100644
--- a/pkg/sdk/reports_test.go
+++ b/pkg/sdk/reports_test.go
@@ -242,7 +242,7 @@ func TestViewReportConfig(t *testing.T) {
 				tc.session = smqauthn.Session{DomainUserID: domainID + "_" + validID, UserID: validID, DomainID: domainID}
 			}
 			authCall := auth.On("Authenticate", mock.Anything, tc.token).Return(tc.session, tc.authenticateErr)
-			svcCall := rsvc.On("ViewReportConfig", mock.Anything, tc.session, tc.id).Return(tc.svcRes, tc.svcErr)
+			svcCall := rsvc.On("ViewReportConfig", mock.Anything, tc.session, tc.id, mock.Anything).Return(tc.svcRes, tc.svcErr)
 			result, err := mgsdk.ViewReportConfig(context.Background(), tc.id, domainID, tc.token)
 			assert.Equal(t, tc.wantErr, err != nil)
 			if !tc.wantErr {
diff --git a/pkg/sdk/rules_test.go b/pkg/sdk/rules_test.go
index fc04022fd..7af20d872 100644
--- a/pkg/sdk/rules_test.go
+++ b/pkg/sdk/rules_test.go
@@ -160,7 +160,7 @@ func TestViewRule(t *testing.T) {
 				tc.session = smqauthn.Session{DomainUserID: domainID + "_" + validID, UserID: validID, DomainID: domainID}
 			}
 			authCall := auth.On("Authenticate", mock.Anything, tc.token).Return(tc.session, tc.authenticateErr)
-			svcCall := rsvc.On("ViewRule", mock.Anything, tc.session, tc.id).Return(tc.svcRes, tc.svcErr)
+			svcCall := rsvc.On("ViewRule", mock.Anything, tc.session, tc.id, mock.Anything).Return(tc.svcRes, tc.svcErr)
 			result, err := mgsdk.ViewRule(context.Background(), tc.id, domainID, tc.token)
 			assert.Equal(t, tc.wantErr, err != nil)
 			if !tc.wantErr {