diff --git a/test/porousmediumflow/3pwateroil/implicit/CMakeLists.txt b/test/porousmediumflow/3pwateroil/implicit/CMakeLists.txt
index 80d6dcecad7491b00c958a121b8184af53e807e7..3759f6c8b81be8db60e48f6f267ba2812a53568a 100644
--- a/test/porousmediumflow/3pwateroil/implicit/CMakeLists.txt
+++ b/test/porousmediumflow/3pwateroil/implicit/CMakeLists.txt
@@ -6,7 +6,7 @@ dune_add_test(NAME test_3pwateroil_sagd_box
               COMMAND ${CMAKE_SOURCE_DIR}/bin/testing/runtest.py
                 CMD_ARGS --script fuzzy
             --files ${CMAKE_SOURCE_DIR}/test/references/test_3pwateroil_sagd_box-reference.vtu
-                   ${CMAKE_CURRENT_BINARY_DIR}/test_3pwateroil_sagd_box-00011.vtu
+                   ${CMAKE_CURRENT_BINARY_DIR}/test_3pwateroil_sagd_box-00001.vtu
             --command "${CMAKE_CURRENT_BINARY_DIR}/test_3pwateroil_sagd_box params.input -Problem.Name test_3pwateroil_sagd_box"
   )
 
diff --git a/test/porousmediumflow/3pwateroil/implicit/main.cc b/test/porousmediumflow/3pwateroil/implicit/main.cc
index 5144eeda923f110cf58b705142566859ccf3f815..5166b0b2739cdbfc87d095d757003e73e7c0a955 100644
--- a/test/porousmediumflow/3pwateroil/implicit/main.cc
+++ b/test/porousmediumflow/3pwateroil/implicit/main.cc
@@ -181,7 +181,8 @@ int main(int argc, char** argv) try
         timeLoop->setTimeStepSize(nonLinearSolver.suggestTimeStepSize(timeLoop->timeStepSize()));
 
         // write vtk output
-        vtkWriter.write(timeLoop->time());
+        if (timeLoop->isCheckPoint() || timeLoop->finished())
+            vtkWriter.write(timeLoop->time());
 
     } while (!timeLoop->finished());
 
diff --git a/test/porousmediumflow/3pwateroil/implicit/problem.hh b/test/porousmediumflow/3pwateroil/implicit/problem.hh
index 33eeffb4d3277b0642a26a385aa20b82bdeb5d4b..98badeef76350c897bbc48cbd500f03ea9a73419 100644
--- a/test/porousmediumflow/3pwateroil/implicit/problem.hh
+++ b/test/porousmediumflow/3pwateroil/implicit/problem.hh
@@ -317,8 +317,6 @@ private:
     mutable Scalar massProducedWater_;
 
     std::string name_;
-
-    std::ofstream massBalance;
 };
 } //end namespace