diff --git a/script/local.py b/script/local.py
index f3bbfd0..7674a72 100755
--- a/script/local.py
+++ b/script/local.py
@@ -121,7 +121,7 @@ def elogllrf(subject, message, system, subsystem, section, attachments = []):
     print out
 
 
-def get_plot_snapshots(title = None, file_type = "png", temp_path = controller.setup.getContextPath()):
+def get_plot_snapshots(title = None, file_type = "png", temp_path = context.setup.getContextPath()):
     """
     Returns list with file names of plots snapshots from a plotting context.
     """
@@ -184,7 +184,7 @@ scan_start.write(-10.0)
 scan_stop.write(10.0)
 scan_step.write(1.0)
 
-if not controller.localMode:
+if not context.localMode:
     CAS.setServerPort(12345)
     cas1=CAS("PSHELL:scanpos",scan_pos, "double")
     cas2=CAS("PSHELL:scanval",scan_val, "double")