diff --git a/opm/polymer/TransportModelCompressiblePolymer.hpp b/opm/polymer/TransportModelCompressiblePolymer.hpp index aa7820838..5d7c94aed 100644 --- a/opm/polymer/TransportModelCompressiblePolymer.hpp +++ b/opm/polymer/TransportModelCompressiblePolymer.hpp @@ -21,7 +21,7 @@ #define OPM_TRANSPORTMODELCOMPRESSIBLEPOLYMER_HEADER_INCLUDED #include -#include +#include #include #include #include @@ -41,7 +41,7 @@ namespace Opm /// Implements a reordering transport solver for incompressible two-phase flow /// with polymer in the water phase. /// \TODO Include permeability reduction effect. - class TransportModelCompressiblePolymer : public TransportModelInterface + class TransportModelCompressiblePolymer : public ReorderSolverInterface { public: diff --git a/opm/polymer/TransportModelPolymer.hpp b/opm/polymer/TransportModelPolymer.hpp index c9d3fd5c7..ff1c8fc24 100644 --- a/opm/polymer/TransportModelPolymer.hpp +++ b/opm/polymer/TransportModelPolymer.hpp @@ -21,7 +21,7 @@ #define OPM_TRANSPORTMODELPOLYMER_HEADER_INCLUDED #include -#include +#include #include #include #include @@ -36,7 +36,7 @@ namespace Opm /// Implements a reordering transport solver for incompressible two-phase flow /// with polymer in the water phase. /// \TODO Include permeability reduction effect. - class TransportModelPolymer : public TransportModelInterface + class TransportModelPolymer : public ReorderSolverInterface { public: