diff --git a/script/local.py b/script/local.py
index 040e84e4..04a3d952 100644
--- a/script/local.py
+++ b/script/local.py
@@ -211,12 +211,15 @@ def append_diag_datasets(parent = None):
         append_dataset(group+get_diag_name(f), x)
 
 
-def before_readout():
+def wait_beam():
     if not beam_ok:
         print "Waiting for beam..."
         while not beam_ok:
             time.sleep(0.1)
         print "Beam ok"
+
+def before_readout():
+    wait_beam()
     trig_scienta()