Fix init highlighting