diff --git a/tensorflow_probability/python/experimental/autobnn/BUILD b/tensorflow_probability/python/experimental/autobnn/BUILD
index a719518a8b..6e851b715a 100644
--- a/tensorflow_probability/python/experimental/autobnn/BUILD
+++ b/tensorflow_probability/python/experimental/autobnn/BUILD
@@ -175,6 +175,7 @@ py_library(
 
 py_test(
     name = "models_test",
+    timeout = "long",
     srcs = ["models_test.py"],
     shard_count = 3,
     deps = [