Fix some weird bugs in Lex visualisation