diff --git a/src/language/utils/PugsFunctionAdapter.hpp b/src/language/utils/PugsFunctionAdapter.hpp
index 3047652e3017fb2cd193b799a8edf454e3e4c22b..d39e0f4233df70709d3d765146447727f7541ae4 100644
--- a/src/language/utils/PugsFunctionAdapter.hpp
+++ b/src/language/utils/PugsFunctionAdapter.hpp
@@ -6,9 +6,9 @@
 #include <language/utils/ASTNodeDataType.hpp>
 #include <language/utils/ASTNodeDataTypeTraits.hpp>
 #include <language/utils/SymbolTable.hpp>
-#include <utils/Array.hpp>
 #include <utils/Exceptions.hpp>
 #include <utils/PugsMacros.hpp>
+#include <utils/SmallArray.hpp>
 
 #include <Kokkos_Core.hpp>
 
@@ -128,7 +128,7 @@ class PugsFunctionAdapter<OutputType(InputType...)>
   [[nodiscard]] PUGS_INLINE static auto
   getContextList(const ASTNode& expression)
   {
-    Array<ExecutionPolicy> context_list(Kokkos::DefaultExecutionSpace::impl_thread_pool_size());
+    SmallArray<ExecutionPolicy> context_list(Kokkos::DefaultExecutionSpace::impl_thread_pool_size());
     auto& context = expression.m_symbol_table->context();
 
     for (size_t i = 0; i < context_list.size(); ++i) {