diff --git a/modules/theme/supported_themes.nix b/modules/theme/supported_themes.nix
index 814b412..e619bce 100644
--- a/modules/theme/supported_themes.nix
+++ b/modules/theme/supported_themes.nix
@@ -12,9 +12,7 @@
 
   tokyonight = {
     setup = {style ? "night"}: ''
-      -- need to set style before colorscheme to apply
-      vim.g.tokyonight_style = '${style}'
-      vim.cmd[[colorscheme tokyonight]]
+      vim.cmd[[colorscheme tokyonight-${style}]]
     '';
     styles = ["day" "night" "storm"];
   };