From 20dd3374935c78cf433e8832a1144660bcf2bd44 Mon Sep 17 00:00:00 2001
From: Serg <redetection@gmail.com>
Date: Thu, 10 May 2018 22:41:44 +0700
Subject: [PATCH] and fix empty keyboard layout widget too

---
 MTMR/CustomButtonTouchBarItem.swift | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/MTMR/CustomButtonTouchBarItem.swift b/MTMR/CustomButtonTouchBarItem.swift
index e24bf4f..af229d1 100644
--- a/MTMR/CustomButtonTouchBarItem.swift
+++ b/MTMR/CustomButtonTouchBarItem.swift
@@ -55,6 +55,7 @@ class CustomButtonTouchBarItem: NSCustomTouchBarItem, NSGestureRecognizerDelegat
     
     private func reinstallButton() {
         let title = button.attributedTitle
+        let image = button.image
         let cell = CustomButtonCell()
         button.cell = cell
         if let color = backgroundColor {
@@ -66,6 +67,7 @@ class CustomButtonTouchBarItem: NSCustomTouchBarItem, NSGestureRecognizerDelegat
             button.bezelStyle = isBordered ? .rounded : .inline
         }
         button.attributedTitle = title
+        button.image = image
         self.view = button
     }