lib/memoist.rb in memoist-0.9.3 vs lib/memoist.rb in memoist-0.10.0

@@ -191,9 +191,11 @@
         elsif protected_method_defined?(unmemoized_method)
           protected method_name
         end
       end
     end
+    # return a chainable method_name symbol if we can
+    method_names.length == 1 ? method_names.first : method_names
   end
 
   class AlreadyMemoizedError < RuntimeError; end
 end