diff --git a/python/gtsam/preamble/basis.h b/python/gtsam/preamble/basis.h
index d07a75f6f..56a07cfdd 100644
--- a/python/gtsam/preamble/basis.h
+++ b/python/gtsam/preamble/basis.h
@@ -10,3 +10,5 @@
  * Without this they will be automatically converted to a Python object, and all
  * mutations on Python side will not be reflected on C++.
  */
+
+#include <pybind11/stl.h>
diff --git a/python/gtsam/tests/test_basis.py b/python/gtsam/tests/test_basis.py
index 944bdd9f3..0df9a80b0 100644
--- a/python/gtsam/tests/test_basis.py
+++ b/python/gtsam/tests/test_basis.py
@@ -25,7 +25,7 @@ class TestBasis(GtsamTestCase):
         noise = gtsam.noiseModel.Unit.Create(1)
         def testBasis(fitter, basis, f=f):
             data = {x: f(x) for x in datax}
-            fit = fitter(N, data, noise)
+            fit = fitter(data, noise, N)
             coeff = fit.parameters()
             interpy = basis.WeightMatrix(N, interpx) @ coeff
             np.testing.assert_almost_equal(interpy, np.array([f(x) for x in interpx]), decimal=7)