diff --git a/Sim/Export/ExportToPython.cpp b/Sim/Export/ExportToPython.cpp
index 42e5f6c012cc3df4f8524c53a8cec621792cb510..9d44447b981569a4bbe7529f56f0750ed1e9663e 100644
--- a/Sim/Export/ExportToPython.cpp
+++ b/Sim/Export/ExportToPython.cpp
@@ -13,13 +13,15 @@
 //  ************************************************************************************************
 
 #include "Sim/Export/ExportToPython.h"
+#include "Base/Py/PyFmt.h"
 #include "Sim/Export/SampleToPython.h"
 #include "Sim/Export/SimulationToPython.h"
 #include "Sim/Simulation/ISimulation.h"
 
 std::string Py::Export::sampleCode(const MultiLayer& sample)
 {
-    return SampleToPython().sampleCode(sample);
+    std::string code = SampleToPython().sampleCode(sample);
+    return "import bornagain as ba\n" + Py::Fmt::printImportedSymbols(code) + "\n\n" + code;
 }
 
 std::string Py::Export::simulationPlotCode(const ISimulation& simulation)