JavaScript Swap Menu Logo 4 Divi

<script>
    var imageUrl = [
  '/wp-content/uploads/Logo_primary.svg',
  '/wp-content/uploads/Logo_fixed.svg',
  '/wp-content/uploads/Logo_mobile_wide.svg',
  '/wp-content/uploads/Logo_mobile.svg',
    ];
  
    jQuery(window).on('load scroll resize', function() {
        var $header = jQuery('header');
        var $logo = jQuery('#logo');

		if (window.matchMedia('(max-width: 980px)').matches) {
			if (window.matchMedia('(max-width: 724px)').matches) {
    	        return $logo.attr('src', imageUrl[3]);
			};
			return $logo.attr('src', imageUrl[2])
        };
	        if ($header.hasClass('et-fixed-header')) {
	            return $logo.attr('src', imageUrl[1]);
			};
			return $logo.attr('src', imageUrl[0])
    });
</script>