diff --git a/src/aws/restart.sh b/src/aws/restart.sh
index af3bf7f4..7203b086 100644
--- a/src/aws/restart.sh
+++ b/src/aws/restart.sh
@@ -23,9 +23,11 @@ function restart(){
     echo "get the new ip address. The procedure might take time for a while"
     publicIp=$(waitForOutput "getLatestIpAddress $instanceId")
 
+    portConfigArray=$(getArrayValueAsStringByKey $instanceName tcpPortArray)
+
     echo "the new ip address is $publicIp"
     updateIPAddress $balloonName $publicIp
 
     echo "open the new sshtunnel"
-    openSSHTunnel $publicIp
+	openSSHTunnel $publicIp $portConfigArray
 }
\ No newline at end of file