diff --git a/examples/polymer_reorder.cpp b/examples/polymer_reorder.cpp
index cec538c03..340e8464c 100644
--- a/examples/polymer_reorder.cpp
+++ b/examples/polymer_reorder.cpp
@@ -369,7 +369,7 @@ main(int argc, char** argv)
         // Gravity.
         gravity[2] = deck.hasField("NOGRAV") ? 0.0 : Opm::unit::gravity;
         // Init state variables (saturation and pressure).
-        initStateTwophaseFromDeck(*grid->c_grid(), *props, deck, gravity[2], state);
+        initStateFromDeck(*grid->c_grid(), *props, deck, gravity[2], state);
         // Init polymer properties.
         polyprop.readFromDeck(deck);
     } else {
@@ -393,7 +393,7 @@ main(int argc, char** argv)
         // Gravity.
         gravity[2] = param.getDefault("gravity", 0.0);
         // Init state variables (saturation and pressure).
-        initStateTwophaseBasic(*grid->c_grid(), *props, param, gravity[2], state);
+        initStateBasic(*grid->c_grid(), *props, param, gravity[2], state);
         // Init polymer properties.
         // Setting defaults to provide a simple example case.
         double c_max = param.getDefault("c_max_limit", 5.0);
diff --git a/opm/polymer/PolymerState.hpp b/opm/polymer/PolymerState.hpp
index 6d49aa27f..99488abe3 100644
--- a/opm/polymer/PolymerState.hpp
+++ b/opm/polymer/PolymerState.hpp
@@ -34,9 +34,9 @@ namespace Opm
     class PolymerState : public TwophaseState
     {
     public:
-        void init(const UnstructuredGrid& g)
+        void init(const UnstructuredGrid& g, int num_phases)
         {
-            TwophaseState::init(g);
+            TwophaseState::init(g, num_phases);
             concentration_.resize(g.number_of_cells, 0.0);
             cmax_.resize(g.number_of_cells, 0.0);
         }