fix early stop bug