fix return datatype