fix return value bug\