fix return value type