diff --git a/assets/js/gp-main.js b/assets/js/gp-main.js
index b639d42..0bb6fa8 100644
--- a/assets/js/gp-main.js
+++ b/assets/js/gp-main.js
@@ -2,8 +2,8 @@
 
 // header
 toggleOpenX('.lang', '.lang__open', '.lang__list', '.lang__content', false);
-singleToggleOpenX('#pc-menu','.header__menu-block','.header__pc-menu', '.header', 'white');
-singleToggleOpenX('#phone-menu','.header__menu-block','.header__phone-menu', '.header', 'white');
+toggleHeader('#pc-menu','.header__menu-block','.header__pc-menu', '.white', 'white');
+toggleHeader('#phone-menu','.header__menu-block','.header__phone-menu', '.white', 'white');
 // header 
 
 // media
@@ -249,14 +249,11 @@ function toggleOpenX(mainElement, buttonElement ,heightElement, contentElement,
     });
 }
 
-function singleToggleOpenX(button, content, blockheight, removeBlock, removeClass) {
+function toggleHeader(button, content, blockheight, removeBlock, removeClass) {
     let thisButton = document.querySelector(button), 
         thisContent = document.querySelector(content),
+        thisRemoveBlock = document.querySelector(removeBlock) || '',
         thisBlockheight = document.querySelector(blockheight);
-    
-    // if (removeBlock) {
-    //     let thisRemoveBlock = ;
-    // }
 
     thisButton.onclick = function () {
         let height = thisBlockheight.clientHeight;
@@ -266,14 +263,14 @@ function singleToggleOpenX(button, content, blockheight, removeBlock, removeClas
             thisContent.classList .add('open');
 
             if (removeBlock) {
-                document.querySelector(removeBlock).classList.remove(removeClass);
+                thisRemoveBlock.classList.remove(removeClass);
             }
         }else{
             thisContent.style.height = null;
             thisContent.classList .remove('open');
 
             if (removeBlock) {
-                document.querySelector(removeBlock).classList.add(removeClass);
+                thisRemoveBlock.classList.add(removeClass);
             }
         }
     }