fix return bug