Fixes an issue when an error is raised in a Timecop.return block, whi…

…ch unsets any global freezes or travels

lib/timecop/timecop.rb

@@ -150,6 +150,7 @@ def return(&block)
     current_baseline = @baseline
     unmock!
     yield
+  ensure
     @_stack = current_stack
     @baseline = current_baseline
   end

test/timecop_test.rb

@@ -219,6 +219,14 @@ def test_exception_thrown_in_freeze_block_properly_resets_time
     end
   end
 
+  def test_exception_thrown_in_return_block_restores_previous_time
+    t = Time.local(2008, 10, 10, 10, 10, 10)
+    Timecop.freeze(t) do
+      Timecop.return { raise 'foobar' } rescue nil
+      assert_equal t, Time.now
+    end
+  end
+
   def test_freeze_freezes_time
     t = Time.local(2008, 10, 10, 10, 10, 10)
     now = Time.now