fixed a nasty bug with the method highlighting