Fix text highlighting again