diff --git a/tensorflow/python/keras/BUILD b/tensorflow/python/keras/BUILD
index 52ed1cbda56..0584d1ca97f 100755
--- a/tensorflow/python/keras/BUILD
+++ b/tensorflow/python/keras/BUILD
@@ -617,7 +617,7 @@ tf_py_test(
 
 tf_py_test(
     name = "data_adapter_test",
-    size = "small",
+    size = "medium",
     srcs = ["engine/data_adapter_test.py"],
     additional_deps = [
         ":data_adapter",