diff --git a/omnn/extrapolator/Extrapolator.cpp b/omnn/extrapolator/Extrapolator.cpp
index 176bcad47..b00d72952 100644
--- a/omnn/extrapolator/Extrapolator.cpp
+++ b/omnn/extrapolator/Extrapolator.cpp
@@ -16,9 +16,9 @@ using namespace boost::numeric::ublas;
 // Replace incorrect allocator functions with correct Boost Ublas container functions
 
 // Ensure that matrix_t, vector_t, and permutation_matrix_t are using unbounded_array_wrapper with custom_allocator
-using matrix_t = boost::numeric::ublas::matrix<Valuable, boost::numeric::ublas::basic_row_major<>, unbounded_array_wrapper<Valuable>>;
-using vector_t = boost::numeric::ublas::vector<Valuable, unbounded_array_wrapper<Valuable>>;
-using permutation_matrix_t = boost::numeric::ublas::permutation_matrix<std::size_t, unbounded_array_wrapper<std::size_t>>;
+using matrix_t = boost::numeric::ublas::matrix<Valuable, boost::numeric::ublas::basic_row_major<>, unbounded_array_wrapper<Valuable, custom_allocator<Valuable>>>;
+using vector_t = boost::numeric::ublas::vector<Valuable, unbounded_array_wrapper<Valuable, custom_allocator<Valuable>>>;
+using permutation_matrix_t = boost::numeric::ublas::permutation_matrix<std::size_t, unbounded_array_wrapper<std::size_t, custom_allocator<std::size_t>>>;
 
 // This function calculates the determinant of a matrix using LU factorization
 auto det_fast(matrix_t matrix) {