lib/anycable/cli.rb in anycable-1.0.2 vs lib/anycable/cli.rb in anycable-1.0.3

@@ -101,12 +101,12 @@
 
     def logger
       AnyCable.logger
     end
 
-    def at_stop
-      if block_given?
-        @at_stop << Proc.new
+    def at_stop(&block)
+      if block
+        @at_stop << block
       else
         @at_stop
       end
     end