diff --git a/pkg/state/BUILD b/pkg/state/BUILD
index 012b0484e..6a5b2d4ff 100644
--- a/pkg/state/BUILD
+++ b/pkg/state/BUILD
@@ -71,7 +71,7 @@ go_proto_library(
 
 go_test(
     name = "state_test",
-    size = "small",
+    timeout = "long",
     srcs = ["state_test.go"],
     embed = [":state"],
 )