diff --git a/bin/omakub-theme b/bin/omakub-theme
index 69e4790..544199b 100755
--- a/bin/omakub-theme
+++ b/bin/omakub-theme
@@ -14,9 +14,9 @@ sed -i "s/theme \".*\"/theme \"$THEME\"/g" ~/.config/zellij/config.kdl
 if [ "$THEME" == "gruvbox" ]; then
 	VSC_THEME="Gruvbox Dark Medium"
 elif [ "$THEME" == "catppuccin" ]; then
-else
-	VSC_THEME=$THEME
 	VSC_THEME="Catppuccin Macchiato"
+elif [ "$THEME" == "tokyo-night" ]; then
+	VSC_THEME="Tokyo Night"
 fi
 sed -i "s/\"workbench.colorTheme\": \".*\"/\"workbench.colorTheme\": \"$VSC_THEME\"/g" ~/.config/Code/User/settings.json