Fixes how the current class is applied