diff --git a/javatests/google/registry/monitoring/whitebox/BUILD b/javatests/google/registry/monitoring/whitebox/BUILD
index 76022fa85..c8e4563e4 100644
--- a/javatests/google/registry/monitoring/whitebox/BUILD
+++ b/javatests/google/registry/monitoring/whitebox/BUILD
@@ -43,6 +43,7 @@ java_library(
 
 GenTestRules(
     name = "GeneratedTestRules",
+    medium_tests = ["VerifyEntityIntegrityActionTest"],
     test_files = glob(["*Test.java"]),
     deps = [":whitebox"],
 )