Fix signature highlighting