Add the missing injection in the run function.