lib/toycol/protocol.rb in toycol-0.2.2 vs lib/toycol/protocol.rb in toycol-0.3.0

@@ -15,10 +15,10 @@
       def define(protocol_name = nil, &block)
         @definements[protocol_name] = block
       end
 
       # For application which use the protocol
-      def use(protocol_name)
+      def use(protocol_name = nil)
         @protocol_name = protocol_name
       end
 
       # For server which use the protocol
       def run!(message)