diff --git a/karabo_devices/shared_mem_processor/src/onlinemid_karabo/processor.py b/karabo_devices/shared_mem_processor/src/onlinemid_karabo/processor.py
index 8bac75b4b12872213b2195ddcf61b753607dfeca..74bb63853ba515c4343d3224902495c5a4dca5c1 100644
--- a/karabo_devices/shared_mem_processor/src/onlinemid_karabo/processor.py
+++ b/karabo_devices/shared_mem_processor/src/onlinemid_karabo/processor.py
@@ -185,8 +185,9 @@ class OnlineMIDProcessor(TrainMatcher.TrainMatcher):
 
         
         self.updateSchema(self.onlineImagePlotter.get_schema_update("plots", 'first_bunch', AgipdModulePlotter))
-        #self.updateSchema(self.onlineImagePlotter.get_schema_update("plots", 'gain_stage', GainStagePlotter))
-
+        self.assembled_output = self.signalSlotable.getOutputChannel("plots")
+        
+        self.updateSchema(self.onlineImagePlotter.get_schema_update("plots", 'gain_stage', GainStagePlotter))
         self.assembled_output = self.signalSlotable.getOutputChannel("plots")
         
         if not self.get("enableKaraboOutput"):
@@ -379,7 +380,7 @@ class OnlineMIDProcessor(TrainMatcher.TrainMatcher):
         plot_hash = Hash()
 
         plot_hash = self.onlineImagePlotter.add_to_hash(plot_hash, 'first_bunch', train_id, detector_data_list[0][1][0])
-        #plot_hash = self.onlineImagePlotter.add_to_hash(plot_hash, 'gain_stage', train_id, numPixelPerGainStage)
+        plot_hash = self.onlineImagePlotter.add_to_hash(plot_hash, 'gain_stage', train_id, numPixelPerGainStage)