diff --git a/modules/programs/kubecolor.nix b/modules/programs/kubecolor.nix
index 6ab66375f..807a55550 100644
--- a/modules/programs/kubecolor.nix
+++ b/modules/programs/kubecolor.nix
@@ -92,7 +92,10 @@ in {
 
     home.shellAliases = lib.mkIf (cfg.enableAlias && (cfg.package != null)) {
       kubectl = lib.getExe cfg.package;
-      oc = lib.mkIf (builtins.elem pkgs.openshift config.home.packages) "env KUBECTL_COMMAND=${lib.getExe pkgs.openshift} ${lib.getExe cfg.package}";
+      oc = lib.mkIf (builtins.elem pkgs.openshift config.home.packages)
+        "env KUBECTL_COMMAND=${lib.getExe pkgs.openshift} ${
+          lib.getExe cfg.package
+        }";
     };
 
     programs.zsh.initContent =
diff --git a/tests/modules/programs/kubecolor/default.nix b/tests/modules/programs/kubecolor/default.nix
index 4445de604..fe8404265 100644
--- a/tests/modules/programs/kubecolor/default.nix
+++ b/tests/modules/programs/kubecolor/default.nix
@@ -2,4 +2,6 @@
   kubecolor-empty-config = ./empty-config.nix;
   kubecolor-example-config-default-paths = ./example-config-default-paths.nix;
   kubecolor-example-config-xdg-paths = ./example-config-xdg-paths.nix;
+  kubecolor-does-have-openshift = ./does-have-openshift.nix;
+  kubecolor-does-not-have-openshift = ./does-not-have-openshift.nix;
 }
diff --git a/tests/modules/programs/kubecolor/does-have-openshift.nix b/tests/modules/programs/kubecolor/does-have-openshift.nix
new file mode 100644
index 000000000..015468865
--- /dev/null
+++ b/tests/modules/programs/kubecolor/does-have-openshift.nix
@@ -0,0 +1,33 @@
+{ config, pkgs, ... }:
+
+{
+  programs.kubecolor = {
+    enable = true;
+    package = config.lib.test.mkStubPackage {
+      name = "kubecolor";
+      version = "0.4.0";
+    };
+    enableAlias = true;
+  };
+  programs.zsh = {
+    enable = true;
+    package = config.lib.test.mkStubPackage {
+      name = "zsh";
+      version = "5.9";
+    };
+  };
+  nixpkgs.overlays = [
+    (self: super: rec {
+      openshift = config.lib.test.mkStubPackage {
+        name = "openshift";
+        version = "4.16.0";
+      };
+    })
+  ];
+  home.packages = [ pkgs.openshift ];
+
+  nmt.script = ''
+    assertFileRegex 'home-files/.zshrc' '^alias.* oc=.*'
+  '';
+}
+
diff --git a/tests/modules/programs/kubecolor/does-not-have-openshift.nix b/tests/modules/programs/kubecolor/does-not-have-openshift.nix
new file mode 100644
index 000000000..1f4cb0d7c
--- /dev/null
+++ b/tests/modules/programs/kubecolor/does-not-have-openshift.nix
@@ -0,0 +1,24 @@
+{ config, ... }:
+
+{
+  programs.kubecolor = {
+    enable = true;
+    package = config.lib.test.mkStubPackage {
+      name = "kubecolor";
+      version = "0.4.0";
+    };
+    enableAlias = true;
+  };
+  programs.zsh = {
+    enable = true;
+    package = config.lib.test.mkStubPackage {
+      name = "zsh";
+      version = "5.9";
+    };
+  };
+
+  nmt.script = ''
+    assertFileNotRegex 'home-files/.zshrc' '^alias.* oc=.*'
+  '';
+}
+