modifed the injection code