diff --git a/DESCRIPTION b/DESCRIPTION
index de3a26aa..b181cfa1 100644
--- a/DESCRIPTION
+++ b/DESCRIPTION
@@ -22,7 +22,7 @@ URL: https://mlr3learners.mlr-org.com,
     https://github.com/mlr-org/mlr3learners
 BugReports: https://github.com/mlr-org/mlr3learners/issues
 Depends:
-    mlr3 (>= 0.12.0-9000),
+    mlr3 (>= 0.13.1),
     R (>= 3.1.0)
 Imports:
     checkmate,
diff --git a/R/helpers.R b/R/helpers.R
index 33cdba98..03dc356c 100644
--- a/R/helpers.R
+++ b/R/helpers.R
@@ -8,7 +8,7 @@ pvec2mat = function(p, levs) {
 
 
 ordered_features = function(task, learner) {
-  cols = names(learner$state$task_prototype %??% learner$state$data_prototype)
+  cols = names(learner$state$data_prototype)
   task$data(cols = intersect(cols, task$feature_names))
 }