Fix source highlight