Fix error in empty menu locations procedure assignment

yithemes · May 22, 2020 · 17c6d55 · 17c6d55
1 parent 47d2254
commit 17c6d55
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/functions.php b/functions.php
@@ -56,12 +56,19 @@ function yith_proteo_setup() {
 		/**
 		 * Fix empty menu locations
 		 */
-		$menu_locations = get_theme_mod( 'nav_menu_locations' );
+		$menu_locations        = get_theme_mod( 'nav_menu_locations' );
+		$update_menu_locations = false;
 		if ( ! isset( $menu_locations['primary'] ) && isset( $menu_locations['menu-1'] ) ) {
-			set_theme_mod( 'nav_menu_locations', array_merge( $menu_locations, array( 'primary' => $menu_locations['menu-1'] ) ) );
+			$menu_locations        = array_merge( $menu_locations, array( 'primary' => $menu_locations['menu-1'] ) );
+			$update_menu_locations = true;
 		}
 		if ( ! isset( $menu_locations['mobile'] ) ) {
-			set_theme_mod( 'nav_menu_locations', array_merge( $menu_locations, array( 'mobile' => $menu_locations['primary'] ) ) );
+			$menu_locations        = array_merge( $menu_locations, array( 'mobile' => $menu_locations['primary'] ) );
+			$update_menu_locations = true;
+		}
+
+		if ( $update_menu_locations ) {
+			set_theme_mod( 'nav_menu_locations', $menu_locations );
 		}
 
 		/*