fix return type.