diff --git a/fake_tunnel_connector/fake_tunnel_connector.rb b/fake_tunnel_connector/fake_tunnel_connector.rb
index aa689f6..015b672 100644
--- a/fake_tunnel_connector/fake_tunnel_connector.rb
+++ b/fake_tunnel_connector/fake_tunnel_connector.rb
@@ -20,7 +20,7 @@ class FakeTunnelConnector
     response = read_socket(tcp_socket)
     location = response[:start_line][:location]
     write_response(tcp_socket, location)
-    tcp_socket.close
+    close_connection(tcp_socket)
   end
 
   def ssl_server
@@ -74,8 +74,13 @@ class FakeTunnelConnector
 
   def write_response(tcp_socket, location)
     routed_response_string = route_request(location)
+    body = ok(routed_response_string)
     tcp_socket.print(ok_headers)
-    tcp_socket.print(ok(routed_response_string))
+    tcp_socket.print(body)
+  end
+
+  def close_connection(tcp_socket)
+    tcp_socket.close
   end
 
   def ok(body = 'Success')
@@ -94,7 +99,7 @@ class FakeTunnelConnector
     elsif location == '/session'
       login
     else
-      ok
+      login
     end
   end