1 files changed, 3 insertions, 2 deletions
diff --git a/lib/clogger/pure.rb b/lib/clogger/pure.rb
index e9a8e6a..50e4f6e 100644
--- a/lib/clogger/pure.rb
+++ b/lib/clogger/pure.rb
@@ -43,12 +43,13 @@ class Clogger
        @body_bytes_sent += Rack::Utils.bytesize(part)
        yield part
      end
-    ensure
-      log(@env, @status, @headers)
+    self
    end
  
    def close
      @body.close if @body.respond_to?(:close)
+    ensure
+      log(@env, @status, @headers)
    end
  
    def reentrant?