fix return data type