Problem with edd_get_discount_id_by_code

function edd_get_discount_id_by_code will return a notice when i enter an invalid discount so can i change this to :

function edd_get_discount_id_by_code( $code ) {
	$discount = edd_get_discount_by_code( $code );

	if( $discount ) {
		return $discount->ID;
	}else{
		return false;
	}
}

Thanks

http://ww.wp.xz.cn/extend/plugins/easy-digital-downloads/