diff --git a/tensorflow_privacy/privacy/keras_models/BUILD b/tensorflow_privacy/privacy/keras_models/BUILD
index 93afc859..ed3e9fc3 100644
--- a/tensorflow_privacy/privacy/keras_models/BUILD
+++ b/tensorflow_privacy/privacy/keras_models/BUILD
@@ -35,6 +35,7 @@ py_test(
 
 py_test(
     name = "dp_keras_model_distributed_test",
+    timeout = "long",
     srcs = ["dp_keras_model_distributed_test.py"],
     tags = [
         "manual",