diff --git a/opm/simulators/wells/WellState.cpp b/opm/simulators/wells/WellState.cpp index 76fc92f8c..cd93cb2de 100644 --- a/opm/simulators/wells/WellState.cpp +++ b/opm/simulators/wells/WellState.cpp @@ -774,7 +774,7 @@ void WellState::shutWell(int well_index) ws.shut(); } -void WellState::updateStatus(int well_index, Well::Status status) +void WellState::updateStatus(int well_index, WellStatus status) { auto& ws = this->well(well_index); ws.updateStatus(status); diff --git a/opm/simulators/wells/WellState.hpp b/opm/simulators/wells/WellState.hpp index 685b9e7b9..e119ff719 100644 --- a/opm/simulators/wells/WellState.hpp +++ b/opm/simulators/wells/WellState.hpp @@ -34,7 +34,6 @@ #include #include -#include #include #include @@ -51,6 +50,7 @@ namespace Opm class ParallelWellInfo; class Schedule; +enum class WellStatus; /// The state of a set of wells, tailored for use by the fully /// implicit blackoil simulator. @@ -211,7 +211,7 @@ public: bool wellIsOwned(const std::string& wellName) const; - void updateStatus(int well_index, Well::Status status); + void updateStatus(int well_index, WellStatus status); void openWell(int well_index); void shutWell(int well_index);