diff --git a/script/HoloScan.py b/script/HoloScan.py
index 67b041ec..424abeab 100644
--- a/script/HoloScan.py
+++ b/script/HoloScan.py
@@ -10,6 +10,6 @@ RELATIVE (BOOLEAN)
 """
 
 
-#set_preference(Preference.ENABLED_PLOTS, [ManipulatorPhi, ManipulatorTheta, Scienta.dataMatrix, Integration])
-set_preference(Preference.PLOT_TYPES, {Integration:1})
+#set_preference(Preference.ENABLED_PLOTS, [ManipulatorPhi, ManipulatorTheta, Scienta.dataMatrix, ImageIntegrator])
+set_preference(Preference.PLOT_TYPES,{'ImageIntegrator':1})
 ascan((phi, theta), SENSORS, (PHI_RANGE[0], THETA_RANGE[0]), (PHI_RANGE[1], THETA_RANGE[1]), STEPS, LATENCY, RELATIVE, before_read=trig_scienta)
\ No newline at end of file
diff --git a/script/PhotonEnergy.py b/script/PhotonEnergy.py
index 9a61e42c..ff82a4ba 100644
--- a/script/PhotonEnergy.py
+++ b/script/PhotonEnergy.py
@@ -6,7 +6,7 @@ SENSORS (list)
 LATENCY (double)
 """
 
-set_preference(Preference.PLOT_TYPES,{'Integration':1}); 
+set_preference(Preference.PLOT_TYPES,{'ImageIntegrator':1})
 
 if len(VECTOR[0]) == 2:
     #FIXED