fix small bug for not return