diff --git a/modules/home-manager/shell.nix b/modules/home-manager/shell.nix
index 2efd4ac..ad35c12 100644
--- a/modules/home-manager/shell.nix
+++ b/modules/home-manager/shell.nix
@@ -1,5 +1,5 @@
 { config, lib, pkgs, inputs, ... }:
- 
+
 {
   # configure z-shell to use omz with some plugins
   programs.zsh = {
@@ -8,21 +8,21 @@
     syntaxHighlighting.enable = true;
     oh-my-zsh = {
       enable = true;
-      # theme will be set below depending on the shell
-      theme = "";
-      # make omz faster on large repositories
-      custom = "DISABLE_UNTRACKED_FILES_DIRTY=true";
       # add git and agents as plugins
       plugins = [ "git" "ssh-agent" "gpg-agent" ];
+      # everything here lands before `source $ZSH/oh-my-zsh.sh`
+      extraConfig = ''
+        # make omz faster on large repositories
+        DISABLE_UNTRACKED_FILES_DIRTY=true
+
+        # only display a fancy theme when glyphs are rendered
+        if [[ "$TERM" == "linux" || "$TERM" == "screen" ]]; then
+          ZSH_THEME="gentoo"
+        else
+          ZSH_THEME="agnoster"
+        fi
+      '';
     };
-    # only display a fancy theme when glyphs are rendered
-    initContent = lib.mkOrder 550 ''
-      if [[ "$TERM" == "linux" || "$TERM" == "screen" ]]; then
-        ZSH_THEME="gentoo"
-      else
-        ZSH_THEME="agnoster"
-      fi
-    '';
   };
 
   # set important env vars