Fix parameter passed to then() and remove Task.