diff --git a/Wrap/Swig/libBornAgainBase.i b/Wrap/Swig/libBornAgainBase.i
index 6ebbebd519fb21993cf4b5fd62527bd8b8881988..a3963f3e09d06fe277bf44024dfa36f87971cd8d 100644
--- a/Wrap/Swig/libBornAgainBase.i
+++ b/Wrap/Swig/libBornAgainBase.i
@@ -26,6 +26,7 @@
 #include "Base/Type/ICloneable.h"
 #include "Base/Type/Span.h"
 #include "Base/Vector/RotMatrix.h"
+#include "Base/Py/ArrayWrapper.h"
 %}
 
 %include "heinz/Complex.h"
@@ -42,5 +43,7 @@
 %include "Base/Axis/MakeScale.h"
 %include "Base/Axis/Frame.h"
 
+%include "Base/Py/ArrayWrapper.h"
+
 %template(R3) Vec3<double>;
 %template(C3) Vec3<std::complex<double>>;