Change the argument name

Change the argument name and relate it so there is a dependency injeciton