fix return type, miss an ok