diff --git a/dumux/implicit/staggered/model.hh b/dumux/implicit/staggered/model.hh
index 30a874dbc39072c2c28dcbcba1e8a789981b13c4..a62aa33a35531888fa9481a6e85c315a704426be 100644
--- a/dumux/implicit/staggered/model.hh
+++ b/dumux/implicit/staggered/model.hh
@@ -270,10 +270,17 @@ public:
         }
     }
 
+    /*!
+    * \brief Returns the element solution
+    *
+    * \param element The element
+    * \param sol The solution vector
+    * \NOTE: Only returns cell-center related values. Might be revised if face data are needed as well.
+    */
     ElementSolution elementSolution(const Element& element, const SolutionVector& sol) const
     {
         PrimaryVariables priVars(0.0);
-        priVars[0] = sol[cellCenterIdx][this->elementMapper().index(element)];
+        priVars[cellCenterIdx] = sol[cellCenterIdx][this->elementMapper().index(element)];
         return ElementSolution{std::move(priVars)};
     }