lib/proxymachine.rb in proxymachine-1.0.0 vs lib/proxymachine.rb in proxymachine-1.1.0

@@ -1,8 +1,9 @@
 require 'rubygems'
 require 'eventmachine'
 require 'logger'
+require 'socket'
 
 require 'proxymachine/client_connection'
 require 'proxymachine/server_connection'
 
 LOGGER = Logger.new(STDOUT)
@@ -77,10 +78,10 @@
     @@listen = "#{host}:#{port}"
     self.update_procline
     EM.epoll
 
     EM.run do
-      EventMachine::Protocols::ClientConnection.start(host, port)
+      ProxyMachine::ClientConnection.start(host, port)
       trap('QUIT') do
         self.graceful_shutdown('QUIT')
       end
       trap('TERM') do
         self.fast_shutdown('TERM')
\ No newline at end of file