diff --git a/test/com/google/enterprise/adaptor/HeartbeatHandlerTest.java b/test/com/google/enterprise/adaptor/HeartbeatHandlerTest.java
index 91122976..61857a9a 100644
--- a/test/com/google/enterprise/adaptor/HeartbeatHandlerTest.java
+++ b/test/com/google/enterprise/adaptor/HeartbeatHandlerTest.java
@@ -288,11 +288,12 @@ public void getDocContent(Request request, Response response)
         .setAdaptor(mockAdaptor)
         .setAuthzAuthority(mockAdaptor)
         .setWatchdog(watchdog)
-        .setHeaderTimeoutMillis(100)
-        .setContentTimeoutMillis(100)
+        .setHeaderTimeoutMillis(150)
+        .setContentTimeoutMillis(150)
         .build();
     HeartbeatHandler handler = createHeartbeatHandlerBuilder()
         .setWatchdog(watchdog2).setDocHandler(docHandler)
+        .setTimeoutMillis(150)
         .build();
     try {
       handler.handle(ex);