diff --git a/server/store/allocator_sql_test.go b/server/store/allocator_sql_test.go
index f9814c51..da7f54e1 100644
--- a/server/store/allocator_sql_test.go
+++ b/server/store/allocator_sql_test.go
@@ -9,9 +9,11 @@ import (
 
 func TestAllocatorSQLClaimsAndAuditsCompatibleReadyServers(t *testing.T) {
 	for query, fragments := range map[string][]string{
-		RegisterReadyServerSQL: {"game_servers", "ON CONFLICT", "state = 'READY'"},
-		ClaimReadyServerSQL:    {"state = 'READY'", "region = $1", "protocol_version = $3", "FOR UPDATE SKIP LOCKED", "ORDER BY server_id"},
-		InsertAllocationSQL:    {"allocations", "request_digest", "state", "ALLOCATED"},
+		RegisterReadyServerSQL:      {"game_servers", "ON CONFLICT", "state = 'READY'"},
+		ClaimReadyServerSQL:         {"state = 'READY'", "region = $1", "protocol_version = $3", "FOR UPDATE SKIP LOCKED", "ORDER BY server_id"},
+		InsertAllocationSQL:         {"allocations", "request_digest", "state", "ALLOCATED"},
+		ProviderServerClaimSQL:      {"state = 'READY'", "region = $2", "protocol_version = $4", "RETURNING"},
+		ServerAllocationConflictSQL: {"server_id = $1", "FOR UPDATE"},
 	} {
 		for _, fragment := range fragments {
 			if !contains(query, fragment) {