diff --git a/debug.py b/debug.py
index befc0a30a..b7a420c94 100644
--- a/debug.py
+++ b/debug.py
@@ -130,5 +130,8 @@ def profile(fn):
         retval = prof.runcall(fn, *args, **kwargs)
         prof.dump_stats(datafn)
         return retval
-    return wrapper
+    if debug:
+        return decorator(wrapper, fn)
+    else:
+        return fn
     
\ No newline at end of file