fix return value