diff --git a/api4/insights_test.go b/api4/insights_test.go
index 5830762368..73b9c87496 100644
--- a/api4/insights_test.go
+++ b/api4/insights_test.go
@@ -848,7 +848,7 @@ func TestGetTopDMsForUserSince(t *testing.T) {
 
 	// bot creation with permission
 	th.AddPermissionToRole(model.PermissionCreateBot.Id, model.TeamUserRoleId)
-	th.App.UpdateUserRoles(th.BasicUser.Id, model.TeamUserRoleId+" "+model.SystemUserRoleId, false)
+	th.App.UpdateUserRoles(th.Context, th.BasicUser.Id, model.TeamUserRoleId+" "+model.SystemUserRoleId, false)
 	bot := &model.Bot{
 		Username:    GenerateTestUsername(),
 		DisplayName: "a bot",