Added some injection