diff --git a/fusil/process/replay_python.py b/fusil/process/replay_python.py
index ad501e0..c020d49 100644
--- a/fusil/process/replay_python.py
+++ b/fusil/process/replay_python.py
@@ -88,9 +88,8 @@ def debug(self, level, format, *args):
 
     def writeFunction(self, name, callback, *args):
         self.write(0, "def %s:" % name)
-        old = self.addLevel(1)
-        callback(*args)
-        self.restoreLevel(old)
+        with self.indented():
+            callback(*args)
         self.emptyLine()
 
     def pythonImports(self):