Fix for some of the injection code.