diff --git a/dotfiles/.config/waybar/config b/dotfiles/.config/waybar/config
index 4312297..207aefd 100644
--- a/dotfiles/.config/waybar/config
+++ b/dotfiles/.config/waybar/config
@@ -4,7 +4,7 @@
   "height": 30,
   "modules-left": ["sway/workspaces", "sway/mode", "sway/window"],
   "modules-center": [],
-  "modules-right": ["temperature", "network", "battery", "pulseaudio", "backlight", "cpu", "memory", "custom/layout", "clock", "tray"],
+  "modules-right": ["temperature", "network", "battery", "pulseaudio", "backlight", "cpu", "memory", "clock", "sway/language", "tray"],
   // Module config
   "sway/mode": {
     "tooltip": true,
@@ -73,10 +73,8 @@
   "memory": {
     "format": " {}%"
   },
-  "custom/layout": {
-    "exec": "~/.config/waybar/keyboard-layout.py",
-    "return-type": "json",
-    "format": "| <span color='#ef9f33'>{}</span>"
+  "sway/language": {
+	"format": "| <span color='#bdbd45'>{short} {variant}</span>",
   },
   "clock": {
     "tooltip": false,