lib/memento/action_controller_methods.rb in memento-0.3.7 vs lib/memento/action_controller_methods.rb in memento-0.4.0

@@ -1,10 +1,10 @@
-class Memento
+module Memento
   module ActionControllerMethods
-    
+
     def memento
       block_result = nil
-      memento_session = Memento.instance.memento(current_user) do
+      memento_session = Memento(current_user) do
         block_result = yield
       end
       if memento_session
         response.headers["X-Memento-Session-Id"] = memento_session.id.to_s
       end
\ No newline at end of file