Fix mistake
Use highlighting to detect comments instead of hardcoded tokens