Fix injection bug