When we just add the classes, we should be able to remove this.