diff --git a/opm/simulators/wells/BlackoilWellModel.hpp b/opm/simulators/wells/BlackoilWellModel.hpp
index cb0b84b7e..16d947077 100644
--- a/opm/simulators/wells/BlackoilWellModel.hpp
+++ b/opm/simulators/wells/BlackoilWellModel.hpp
@@ -389,6 +389,14 @@ template<class Scalar> class WellContributions;
 
             void setupDomains(const std::vector<Domain>& domains);
 
+            const SparseTable<int>& well_local_cells() const
+            {
+                return well_local_cells_;
+            }
+            auto begin() const { return well_container_.begin(); }
+            auto end() const { return well_container_.end(); }
+            bool empty() const { return well_container_.empty(); }
+
         protected:
             Simulator& simulator_;
 
@@ -411,8 +419,9 @@ template<class Scalar> class WellContributions;
             createTypedWellPointer(const int wellID,
                                    const int time_step) const;
 
-            WellInterfacePtr createWellForWellTest(const std::string& well_name, const int report_step, DeferredLogger& deferred_logger) const;
-
+            WellInterfacePtr createWellForWellTest(const std::string& well_name,
+                                                   const int report_step,
+                                                   DeferredLogger& deferred_logger) const;
 
             const ModelParameters param_;
             std::size_t global_num_cells_{};
diff --git a/opm/simulators/wells/BlackoilWellModelGeneric.hpp b/opm/simulators/wells/BlackoilWellModelGeneric.hpp
index 2ef35efd8..ff867ab52 100644
--- a/opm/simulators/wells/BlackoilWellModelGeneric.hpp
+++ b/opm/simulators/wells/BlackoilWellModelGeneric.hpp
@@ -208,6 +208,11 @@ public:
     const std::map<std::string, double>& wellOpenTimes() const { return well_open_times_; }
     const std::map<std::string, double>& wellCloseTimes() const { return well_close_times_; }
 
+    const std::map<std::string, int>& well_domain() const
+    {
+        return well_domain_;
+    }
+
     bool reportStepStarts() const { return report_step_starts_; }
 
     bool shouldBalanceNetwork(const int reportStepIndex,