diff --git a/agent/src/main/java/com/microsoft/hydralab/agent/socket/AgentWebSocketClient.java b/agent/src/main/java/com/microsoft/hydralab/agent/socket/AgentWebSocketClient.java
index 9717bb7d7..91cd5f8a4 100644
--- a/agent/src/main/java/com/microsoft/hydralab/agent/socket/AgentWebSocketClient.java
+++ b/agent/src/main/java/com/microsoft/hydralab/agent/socket/AgentWebSocketClient.java
@@ -76,9 +76,10 @@ public void onClose(int code, String reason, boolean remote) {
         }
 
         // if the connection is closed by server with 1008, wait and retry
+        // Allow up to 10 reconnect attempts before exiting
         if (code == CloseReason.CloseCodes.VIOLATED_POLICY.getCode()) {
             violatedReconnectTime++;
-            if(violatedReconnectTime > 5) {
+            if(violatedReconnectTime > 10) {
                 log.error("onClose, code: {}, reason: {}, remote: {}, reconnectTime: {}, {}", code, reason, remote, reconnectTime, violatedReconnectTime);
                 System.exit(code);
             } else {
diff --git a/gradlew b/gradlew
old mode 100644
new mode 100755