diff --git a/lib/action_profiler.rb b/lib/action_profiler.rb
index 4fc8f3d..05c80e9 100644
--- a/lib/action_profiler.rb
+++ b/lib/action_profiler.rb
@@ -23,7 +23,7 @@ def action_profiler(&block)
 
           response.headers['Content-Length'] = response.body.size
           response.headers['Content-Type'] = 'application/octet-stream'
-          response.headers['Content-Disposition'] = %(attachment; filename="#{File.basename(request.path)}.tree")
+          response.headers['Content-Disposition'] = %(attachment; filename="#{File.basename(request.path)}.#{mode}.tree")
         end
       else
         yield