Simplify injection code