Adding missing inject dependency