diff --git a/logstash/fake-logstash-agent.rb b/logstash/fake-logstash-agent.rb
index 7cb5911..fa2799c 100644
--- a/logstash/fake-logstash-agent.rb
+++ b/logstash/fake-logstash-agent.rb
@@ -2,16 +2,8 @@
 
 require 'socket'
 
-
-def main
-  server = TCPServer.new(55784)
-  loop do
-    client = server.accept
-    $stderr.puts client.gets
-  end
-end
-
-
-if __FILE__ == $0
-  main
+server = TCPServer.new(55784)
+loop do
+  client = server.accept
+  $stderr.puts client.gets
 end