diff --git a/script/local.py b/script/local.py
index e22d986a..9ec7cde7 100644
--- a/script/local.py
+++ b/script/local.py
@@ -170,9 +170,9 @@ def append_diag_datasets():
 
 
 SAVE_DIAGS = True
-def AfterReadout(rec):
+def AfterReadout(rec = None):
     if SAVE_DIAGS:
-        if rec.index == 0:
+        if (rec is None) or (rec.index == 0):
             create_diag_datasets()
         append_diag_datasets()