Remove class thats not needed