diff --git a/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD b/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD
index 04ad5d52642..293dc4864cc 100644
--- a/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD
+++ b/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD
@@ -95,7 +95,7 @@ py_test(
 
 py_test(
     name = "choose_fastest_branch_dataset_serialization_test",
-    size = "small",
+    size = "medium",
     srcs = ["choose_fastest_branch_dataset_serialization_test.py"],
     srcs_version = "PY2AND3",
     tags = [