fix remove class issue