diff --git a/python/ray/train/v2/BUILD.bazel b/python/ray/train/v2/BUILD.bazel
index 3d40b2a1715f..828de81494fb 100644
--- a/python/ray/train/v2/BUILD.bazel
+++ b/python/ray/train/v2/BUILD.bazel
@@ -551,7 +551,7 @@ py_test(
 
 py_test(
     name = "test_util",
-    size = "small",
+    size = "medium",
     srcs = ["tests/test_util.py"],
     env = {"RAY_TRAIN_V2_ENABLED": "1"},
     tags = [