diff --git a/main.c b/main.c
index 8608e15..bae3479 100644
--- a/main.c
+++ b/main.c
@@ -49,8 +49,9 @@ int StartEnv(int port) {
         bytes_received = recv(client_sock, buffer, sizeof(buffer), 0);
         if(bytes_received < 0) {
             printf("Receive Error: %s (%d)\n", strerror(errno), errno);
+            break;
         } else if(bytes_received == 0) {
-            continue;
+            break;
         }
 
         if(buffer[0] == 0x0) { // Step
@@ -80,6 +81,7 @@ int StartEnv(int port) {
     }
     shutdown(client_sock, SHUT_RDWR);
     close(client_sock);
+    return 0;
 }
 
 int main(int argc, char* argv[]) {