Fixed problem with injection