Fix heredocs highlighting