lib/authlogic/session/callbacks.rb in authlogic-1.2.1 vs lib/authlogic/session/callbacks.rb in authlogic-1.2.2

@@ -47,17 +47,17 @@
       #   save
       #   # the following are only run is save is successful
       #   after_save
       #   before_update # only if new_session? == false
       #   before_create # only if new_session? == true
-      def save_with_callbacks
+      def save_with_callbacks(&block)
         run_callbacks(:before_save)
         if new_session?
           run_callbacks(:before_create)
         else
           run_callbacks(:before_update)
         end
-        result = save_without_callbacks
+        result = save_without_callbacks(&block)
         if result
           run_callbacks(:after_save)
           
           if new_session?
             run_callbacks(:after_create)
\ No newline at end of file