quick fix to remove class