diff --git a/tests/lsp-servers.nix b/tests/lsp-servers.nix
index 9fc90567..8504998b 100644
--- a/tests/lsp-servers.nix
+++ b/tests/lsp-servers.nix
@@ -36,6 +36,8 @@ let
       ...
     }:
     let
+      inherit (pkgs.stdenv) hostPlatform;
+
       disabled =
         [
           # DEPRECATED SERVERS
@@ -44,7 +46,7 @@ let
           "bufls"
           "typst_lsp"
         ]
-        ++ lib.optionals (pkgs.stdenv.hostPlatform.system == "aarch64-linux") [
+        ++ lib.optionals (hostPlatform.isLinux && hostPlatform.isAarch64) [
           # pkgs.vectorcode is not available on this platform
           "vectorcode_server"
 
diff --git a/tests/test-sources/plugins/cmp/all-sources.nix b/tests/test-sources/plugins/cmp/all-sources.nix
index c4c98612..c74afb86 100644
--- a/tests/test-sources/plugins/cmp/all-sources.nix
+++ b/tests/test-sources/plugins/cmp/all-sources.nix
@@ -1,4 +1,4 @@
-{ pkgs, ... }:
+{ lib, pkgs, ... }:
 {
   all-sources =
     { config, ... }:
@@ -18,18 +18,24 @@
           settings.sources =
             with pkgs.lib;
             let
-              disabledSources = [
-                # We do not provide the required HF_API_KEY environment variable.
-                "cmp_ai"
-                # Triggers the warning complaining about treesitter highlighting being disabled
-                "otter"
-                # Invokes the `nix` command at startup which is not available in the sandbox
-                "nixpkgs_maintainers"
-                # Needs internet access to download `sm-agent`
-                "supermaven"
-                # Sometimes get auth error
-                "codeium"
-              ] ++ optional (pkgs.stdenv.hostPlatform.system == "aarch64-linux") "cmp_tabnine";
+              inherit (pkgs.stdenv) hostPlatform;
+
+              disabledSources =
+                [
+                  # We do not provide the required HF_API_KEY environment variable.
+                  "cmp_ai"
+                  # Triggers the warning complaining about treesitter highlighting being disabled
+                  "otter"
+                  # Invokes the `nix` command at startup which is not available in the sandbox
+                  "nixpkgs_maintainers"
+                  # Needs internet access to download `sm-agent`
+                  "supermaven"
+                  # Sometimes get auth error
+                  "codeium"
+                ]
+                ++ lib.optionals (hostPlatform.isLinux && hostPlatform.isAarch64) [
+                  "cmp_tabnine"
+                ];
             in
             pipe config.cmpSourcePlugins [
               # All known source names