Fix bug with inject