diff --git a/poisson/FemModule.h b/poisson/FemModule.h
index f721986..0a35e32 100644
--- a/poisson/FemModule.h
+++ b/poisson/FemModule.h
@@ -260,7 +260,7 @@ class FemModule
 
   static ARCCORE_HOST_DEVICE inline void
   _computeElementMatrixTETRA4GPU(CellLocalId icell, IndexedCellNodeConnectivityView cnc,
-                                 ax::VariableNodeReal3InView in_node_coord, Real K_e[12]);
+                                 ax::VariableNodeReal3InView in_node_coord, Real K_e[16]);
 
  private:
 
@@ -445,4 +445,4 @@ _computeElementMatrixTETRA4GPU(CellLocalId icell, IndexedCellNodeConnectivityVie
 /*---------------------------------------------------------------------------*/
 /*---------------------------------------------------------------------------*/
 
-#endif
\ No newline at end of file
+#endif