diff --git a/Device/Beam/Beam.h b/Device/Beam/Beam.h
index f780dab16453b0df9006865287d3405e4c88967b..0407e4ea581d209dd741b286550f52bb2240442a 100644
--- a/Device/Beam/Beam.h
+++ b/Device/Beam/Beam.h
@@ -97,7 +97,6 @@ private:
     //! Deprecated for direct use; prefer factory functions InBeam or UnitBeam.
     Beam(double intensity, double wavelength, double alpha, double phi = 0);
     Beam(std::vector<double> P);
-#endif // SWIG
 
     void precompute();
 
@@ -113,6 +112,7 @@ private:
     R3 m_k;
 
     friend class DepthprobeSimulationTest;
+#endif // SWIG
 };
 
 // Factory functions in global namespace: