diff --git a/inc/vk-css-optimize/config.php b/inc/vk-css-optimize/config.php
index b04889868..184953de6 100644
--- a/inc/vk-css-optimize/config.php
+++ b/inc/vk-css-optimize/config.php
@@ -27,7 +27,7 @@ function vk_blocks_css_tree_shaking_handles( $vk_css_tree_shaking_handles ) {
  *
  * @param array $inidata CSS Tree Shaking Exclude Paramator.
  */
-function vk_blocks_css_tree_shaking_exclude_class( $inidata ) {
+function vk_blocks_css_tree_shaking_js_added_class( $jsaddlist ) {
 	$exclude_classes_array = array(
 		'swiper-pagination-bullet',
 		'swiper-pagination-bullet-active',
@@ -73,11 +73,11 @@ function vk_blocks_css_tree_shaking_exclude_class( $inidata ) {
 		'vk_post-col-xxl-2',
 	);
 
-	$inidata['class'] = array_unique( array_merge( $inidata['class'], $exclude_classes_array ) );
+	$jsaddlist = array_unique( array_merge( $jsaddlist, $exclude_classes_array ) );
 
-	return $inidata;
+	return $jsaddlist;
 }
-add_filter( 'css_tree_shaking_exclude', 'vk_blocks_css_tree_shaking_exclude_class' );
+add_filter( 'css_tree_shaking_js_added_class', 'vk_blocks_css_tree_shaking_js_added_class' );
 
 /**
  * CSS Optimize option default