fixed remove class error