diff --git a/pkg/config-api-provider/test/resources/agent_group_assignment_test.go b/pkg/config-api-provider/test/resources/agent_group_assignment_test.go
index 754590a4..8510958e 100644
--- a/pkg/config-api-provider/test/resources/agent_group_assignment_test.go
+++ b/pkg/config-api-provider/test/resources/agent_group_assignment_test.go
@@ -177,7 +177,7 @@ func TestAgentGroupAssignmentResource(t *testing.T) {
 					util.MockGetAgent(
 						"agent_uid",
 						util.GeneratePaginatedResponse([]map[string]interface{}{util.GenerateAgentResponseModel("agent_uid", "")}),
-						1,
+						2,
 					)
 					util.MockGetAgent(
 						"agent_uid_2",