diff --git a/roomserver/state/v1/state_test.go b/roomserver/state/v1/state_test.go
index 4dc7e52ec..5d47d6b46 100644
--- a/roomserver/state/v1/state_test.go
+++ b/roomserver/state/v1/state_test.go
@@ -19,6 +19,7 @@ package v1
 import (
 	"testing"
 
+	"github.com/matrix-org/dendrite/roomserver/state/shared"
 	"github.com/matrix-org/dendrite/roomserver/types"
 )
 
@@ -45,7 +46,7 @@ func TestFindDuplicateStateKeys(t *testing.T) {
 	}}
 
 	for _, test := range testCases {
-		got := findDuplicateStateKeys(test.Input)
+		got := shared.FindDuplicateStateKeys(test.Input)
 		if len(got) != len(test.Want) {
 			t.Fatalf("Wanted %v, got %v", test.Want, got)
 		}
diff --git a/roomserver/state/v2/state_test.go b/roomserver/state/v2/state_test.go
index 36ce49fce..2bda24915 100644
--- a/roomserver/state/v2/state_test.go
+++ b/roomserver/state/v2/state_test.go
@@ -19,6 +19,7 @@ package v2
 import (
 	"testing"
 
+	"github.com/matrix-org/dendrite/roomserver/state/shared"
 	"github.com/matrix-org/dendrite/roomserver/types"
 )
 
@@ -45,7 +46,7 @@ func TestFindDuplicateStateKeys(t *testing.T) {
 	}}
 
 	for _, test := range testCases {
-		got := findDuplicateStateKeys(test.Input)
+		got := shared.FindDuplicateStateKeys(test.Input)
 		if len(got) != len(test.Want) {
 			t.Fatalf("Wanted %v, got %v", test.Want, got)
 		}