diff --git a/test/boxmodels/2p2c/injectionspatialparameters.hh b/test/boxmodels/2p2c/injectionspatialparameters.hh
index 61f6474a2a094e0e6ef6572aca1a8b98df7a1d41..6c49bfca6295e65526f5fe02c35deb33678eef8b 100644
--- a/test/boxmodels/2p2c/injectionspatialparameters.hh
+++ b/test/boxmodels/2p2c/injectionspatialparameters.hh
@@ -225,10 +225,10 @@ public:
      * \param scvfIdx The local index of the sub-control volume face where
      *                    the matrix heat flux should be calculated
      */
-    void matrixHeatFlux(Vector &heatFlux,
+    void matrixHeatFlux(DimVector &heatFlux,
                         const FluxVariables &fluxVars,
                         const ElementVolumeVariables &elemVolVars,
-                        const Vector &tempGrad,
+                        const DimVector &tempGrad,
                         const Element &element,
                         const FVElementGeometry &fvGeometry,
                         const int faceIdx) const