fix return errror