diff --git a/Waveform/WaveRecAlgs/share/PseudoSimToWaveformRecExample_jobOptions.py b/Waveform/WaveRecAlgs/share/PseudoSimToWaveformRecExample_jobOptions.py
index bdfae43cfc3920eea95470937845cdc9de1bf60c..cbcb5836850f8f1958d5e55e22eca9f6afe1f0e4 100644
--- a/Waveform/WaveRecAlgs/share/PseudoSimToWaveformRecExample_jobOptions.py
+++ b/Waveform/WaveRecAlgs/share/PseudoSimToWaveformRecExample_jobOptions.py
@@ -25,7 +25,8 @@ if __name__ == "__main__":
     ConfigFlags.GeoModel.Align.Dynamic = False
     
     ConfigFlags.Input.Files = [
-        "/bundle/data/FASER/LC_output/BatchOutput/TestBeam/TB.Elec.8.r5.e100.SIM.root"
+        "my.HITS.pool.root"
+        #"/bundle/data/FASER/LC_output/BatchOutput/TestBeam/TB.Elec.8.r5.e100.SIM.root"
         ]
 
     
diff --git a/Waveform/WaveRecTools/WaveRecTools/IPseudoSimToWaveformRecTool.icc b/Waveform/WaveRecTools/WaveRecTools/IPseudoSimToWaveformRecTool.icc
index 9bcbd86c09ad0992032e123f0cd01281e7271e18..9f85c6b020b3efc9d81949cbf914b2a424bda19b 100644
--- a/Waveform/WaveRecTools/WaveRecTools/IPseudoSimToWaveformRecTool.icc
+++ b/Waveform/WaveRecTools/WaveRecTools/IPseudoSimToWaveformRecTool.icc
@@ -21,11 +21,12 @@ StatusCode IPseudoSimToWaveformRecTool::reconstruct(const CONT* hitCollection,
       xAOD::WaveformHit* hit = new xAOD::WaveformHit();
       container->push_back(hit);
 
-      hit->set_channel(id.first);
+      hit->set_id(id.first);
+      hit->set_channel(0);
       hit->set_peak(0);
       hit->set_mean(0);
       hit->set_width(0);
-      hit->set_integral(id.second);  // This should be updated to set the ID when exists
+      hit->set_integral(id.second); 
       hit->set_localtime(0);
       hit->set_raw_peak(0);
       hit->set_raw_integral(0);