Merge branch 'fluo-762'
diff --git a/modules/integration/src/test/java/org/apache/fluo/integration/impl/OracleIT.java b/modules/integration/src/test/java/org/apache/fluo/integration/impl/OracleIT.java
index 91636f7..e543c3c 100644
--- a/modules/integration/src/test/java/org/apache/fluo/integration/impl/OracleIT.java
+++ b/modules/integration/src/test/java/org/apache/fluo/integration/impl/OracleIT.java
@@ -36,7 +36,9 @@
 import org.apache.log4j.Level;
 import org.apache.log4j.Logger;
 import org.apache.thrift.server.THsHaServer;
+import org.junit.Rule;
 import org.junit.Test;
+import org.junit.rules.Timeout;
 
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNull;
@@ -44,6 +46,9 @@
 
 public class OracleIT extends ITBaseImpl {
 
+  @Rule
+  public Timeout globalTimeout = Timeout.seconds(60);
+
   @Test
   public void testRestart() throws Exception {
     OracleClient client = env.getSharedResources().getOracleClient();
@@ -225,10 +230,8 @@
     oserver.stop();
     sleepWhileConnected(oserver);
 
-    int count = 0;
-    while (count < 5 && client.getOracle() != null) {
-      Thread.sleep(1000);
-      count++;
+    while (client.getOracle() != null) {
+      Thread.sleep(100);
     }
 
     assertNull(client.getOracle());