Fixed 'add class' loophole