diff --git a/script/local.py b/script/local.py
index 90075f9e..ad274291 100644
--- a/script/local.py
+++ b/script/local.py
@@ -201,10 +201,16 @@ def append_diag_datasets(parent = None):
 
 
 SAVE_DIAGS = True
-def AfterReadout(rec):
+def after_readout(rec):
     if SAVE_DIAGS:
         if rec.index == 0:
             create_diag_datasets()
         append_diag_datasets()
 
+def adjust_sensors():
+    if SENSORS is not None:
+        # Move integration to end
+        for dev in [Integration, AngleDistribution]:
+            if dev in SENSORS:
+                SENSORS=SENSORS+[SENSORS.pop(SENSORS.index(dev))]
     
\ No newline at end of file