diff --git a/src/program_state.hh b/src/program_state.hh
index a1734168d53414f303eb4ddcb0ed5290892c1c6b..bd6b79d4148ccd86e903484ca94924b4c1e0fb2f 100644
--- a/src/program_state.hh
+++ b/src/program_state.hh
@@ -4,6 +4,8 @@
 #include <dune/common/parametertree.hh>
 
 #include <dune/fufem/boundarypatch.hh>
+#include <dune/tnnmg/nonlinearities/zerononlinearity.hh>
+#include <dune/tnnmg/problem-classes/blocknonlineartnnmgproblem.hh>
 
 #include <dune/tectonic/body.hh>
 
diff --git a/src/sand-wedge.cc b/src/sand-wedge.cc
index adc33009ad7323197a2ef61fdb78b20860b08855..670a9584f7752be86849825543939247df3d8bc9 100644
--- a/src/sand-wedge.cc
+++ b/src/sand-wedge.cc
@@ -44,8 +44,6 @@
 #include <dune/solvers/norms/energynorm.hh>
 #include <dune/solvers/solvers/loopsolver.hh>
 #include <dune/solvers/solvers/solver.hh>
-#include <dune/tnnmg/nonlinearities/zerononlinearity.hh>
-#include <dune/tnnmg/problem-classes/blocknonlineartnnmgproblem.hh>
 #include <dune/tnnmg/problem-classes/convexproblem.hh>
 
 #include <dune/tectonic/myblockproblem.hh>