Fix for running on mlr for first release