diff --git a/src/pybinds/PyHeliosSimulation.cpp b/src/pybinds/PyHeliosSimulation.cpp
index 945430495..76db70439 100644
--- a/src/pybinds/PyHeliosSimulation.cpp
+++ b/src/pybinds/PyHeliosSimulation.cpp
@@ -322,13 +322,15 @@ void PyHeliosSimulation::loadSurvey(
     bool writeWaveform,
     bool calcEchowidth,
     bool fullWaveNoise,
-    bool platformNoiseDisabled
+    bool platformNoiseDisabled,
+    bool writePulse
 ){
     xmlreader->sceneLoader.kdtFactoryType = kdtFactory;
     xmlreader->sceneLoader.kdtNumJobs = kdtJobs;
     xmlreader->sceneLoader.kdtSAHLossNodes = kdtSAHLossNodes;
     survey = xmlreader->load(legNoiseDisabled, rebuildScene);
     survey->scanner->setWriteWaveform(writeWaveform);
+    survey->scanner->setWritePulse(writePulse);
     survey->scanner->setCalcEchowidth(calcEchowidth);
     survey->scanner->setFullWaveNoise(fullWaveNoise);
     survey->scanner->setPlatformNoiseDisabled(platformNoiseDisabled);
diff --git a/src/pybinds/PyHeliosSimulation.h b/src/pybinds/PyHeliosSimulation.h
index eb8e3c14d..59cac82ad 100644
--- a/src/pybinds/PyHeliosSimulation.h
+++ b/src/pybinds/PyHeliosSimulation.h
@@ -402,7 +402,8 @@ class PyHeliosSimulation{
         bool writeWaveform = false,
         bool calcEchowidth = false,
         bool fullWaveNoise = false,
-        bool platformNoiseDisabled = true
+        bool platformNoiseDisabled = true,
+        bool writePulse = false
     );
     void addRotateFilter(
         double q0,