diff --git a/plugins/DelayScan.java b/plugins/DelayScan.java
index 4e517b3..c5fd55b 100644
--- a/plugins/DelayScan.java
+++ b/plugins/DelayScan.java
@@ -55,7 +55,7 @@ public class DelayScan extends Panel {
     }
 
     Path getConfigFile() {
-        return Paths.get(getController().getSetup().getConfigPath(), "energy_scan.properties");
+        return Paths.get(getContext().getSetup().getConfigPath(), "energy_scan.properties");
     }
 
     LinePlotBase scanPlot;
diff --git a/plugins/EnergyScan.java b/plugins/EnergyScan.java
index 2ed5f33..6e9a69d 100644
--- a/plugins/EnergyScan.java
+++ b/plugins/EnergyScan.java
@@ -61,7 +61,7 @@ public class EnergyScan extends Panel {
     }
 
     Path getConfigFile() {
-        return Paths.get(getController().getSetup().getConfigPath(), "energy_scan.properties");
+        return Paths.get(getContext().getSetup().getConfigPath(), "energy_scan.properties");
     }
 
     LinePlotBase scanPlot;
diff --git a/plugins/EnergyScan_TBT.java b/plugins/EnergyScan_TBT.java
index e8eb7a5..89f2fcd 100644
--- a/plugins/EnergyScan_TBT.java
+++ b/plugins/EnergyScan_TBT.java
@@ -61,7 +61,7 @@ public class EnergyScan_TBT extends Panel {
     }
 
     Path getConfigFile() {
-        return Paths.get(getController().getSetup().getConfigPath(), "energy_scan.properties");
+        return Paths.get(getContext().getSetup().getConfigPath(), "energy_scan.properties");
     }
 
     LinePlotBase scanPlot;
diff --git a/plugins/Test.java b/plugins/Test.java
index f4e3d10..375369e 100644
--- a/plugins/Test.java
+++ b/plugins/Test.java
@@ -2,7 +2,7 @@
  * Copyright (c) 2014 Paul Scherrer Institute. All rights reserved.
  */
 
-import ch.psi.pshell.core.Controller;
+import ch.psi.pshell.core.Context;
 import ch.psi.pshell.ui.Panel;
 import ch.psi.pshell.ui.Plugin;
 import ch.psi.utils.State;