fix highlighting again