diff --git a/dumux/io/vtkfunction.hh b/dumux/io/vtkfunction.hh
index 31766112d762f0aaf5eb06edca80b81e82811889..885ffa7e0a663a134306d3f5b441c0190b936780 100644
--- a/dumux/io/vtkfunction.hh
+++ b/dumux/io/vtkfunction.hh
@@ -130,7 +130,7 @@ public:
     VectorP1VTKFunction(const GridView& gridView, const Mapper& mapper, const F& field, const std::string& name, int nComps)
     : field_(field), name_(name), nComps_(nComps), mapper_(mapper)
     {
-        if (field.size()!=(unsigned int)( gridView.size(/*codim*/dim)) )
+        if (field.size()!=(unsigned int)( mapper.size() ))
             DUNE_THROW(Dune::IOError, "VectorP1VTKFunction: size mismatch");
     }
 private: