diff --git a/opm/autodiff/FullyImplicitBlackoilSolver.cpp b/opm/autodiff/FullyImplicitBlackoilSolver.cpp
index eed2126e0..5a7545abb 100644
--- a/opm/autodiff/FullyImplicitBlackoilSolver.cpp
+++ b/opm/autodiff/FullyImplicitBlackoilSolver.cpp
@@ -746,11 +746,7 @@ namespace {
                 bhp_targets[w] = -1e100;
                 rate_targets[w] = well_controls_get_current_target(wc);
                 {
-                    /* 
-                       Would have guessed that this should use the 
-                       well_control_get_current_distr() function instead? 
-                    */
-                    const double * distr = well_controls_iget_distr( wc , 0);
+                    const double * distr = well_controls_get_current_distr( wc );
                     for (int phase = 0; phase < np; ++phase) {
                         rate_distr.insert(w, phase*nw + w) = distr[phase];
                     }
diff --git a/opm/autodiff/ImpesTPFAAD.cpp b/opm/autodiff/ImpesTPFAAD.cpp
index b8ce7dffb..4d023039c 100644
--- a/opm/autodiff/ImpesTPFAAD.cpp
+++ b/opm/autodiff/ImpesTPFAAD.cpp
@@ -384,11 +384,7 @@ namespace {
                 bhp_targets[w] = -1e100;
                 rate_targets[w] = well_controls_get_current_target( wc );
                 {
-                    /* 
-                       Would have guessed that this should use the 
-                       well_control_get_current_distr() function instead? 
-                    */
-                    const double * distr = well_controls_iget_distr( wc , 0 );
+                    const double * distr = well_controls_get_current_distr( wc );
                     for (int phase = 0; phase < np; ++phase) {
                         rate_distr.insert(w, phase*nw + w) = distr[phase];
                     }