fix: wrap elements issue with EL custom header

Codeinwp · Jan 10, 2025 · 1275cb7 · 1275cb7
1 parent 8a98977
commit 1275cb7
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/inc/compatibility/woocommerce.php b/inc/compatibility/woocommerce.php
@@ -284,8 +284,13 @@ public function register_hooks() {
 		add_action( 'admin_footer', array( $this, 'update_woo_width' ) );
 
 		// Wrap content.
-		add_action( 'neve_after_primary_start', array( $this, 'wrap_pages_start' ) );
-		add_action( 'neve_before_primary_end', array( $this, 'wrap_pages_end' ) );
+		if ( function_exists( 'elementor_theme_do_location' ) && elementor_theme_do_location( 'header' ) ) {
+			add_action( 'woocommerce_before_main_content', array( $this, 'wrap_pages_start' ) );
+			add_action( 'woocommerce_after_main_content', array( $this, 'wrap_pages_end' ) );
+		} else {
+			add_action( 'neve_after_primary_start', array( $this, 'wrap_pages_start' ) );
+			add_action( 'neve_before_primary_end', array( $this, 'wrap_pages_end' ) );
+		}
 
 		add_action( 'woocommerce_before_main_content', array( $this, 'wrap_main_content_start' ), 15 );
 		add_action( 'woocommerce_after_main_content', array( $this, 'close_div' ), 15 );