lib/gogo_driver.rb in gogo_driver-0.0.1 vs lib/gogo_driver.rb in gogo_driver-0.0.2

@@ -1,11 +1,14 @@
 require "gogo_driver/version"
+require "gogo_driver/entity"
 
 module GogoDriver
   def init
     @entity = Entity.new
   end
 
   def method_missing(method, *args, &block)
     @entity.respond_to?(method) ? @entity.send(method, *args, &block) : super
+  rescue Errno::ECONNREFUSED
+    @entity = Entity.new
   end
 end