diff --git a/script/local.py b/script/local.py
index 3630950f..4b103dfa 100644
--- a/script/local.py
+++ b/script/local.py
@@ -208,9 +208,8 @@ def append_diag_datasets(parent = None):
         append_dataset(group+get_diag_name(f), x)
 
 
-SAVE_DIAGS = True
 def after_readout(rec):
-    if SAVE_DIAGS and get_context().persist:
+    if get_context().persist:
         if rec.index == 0:
             create_diag_datasets()
         append_diag_datasets()