fixed the early stop removal