fix return error