diff --git a/services/ollama.nix b/services/ollama.nix
index 2578492..4e78c15 100644
--- a/services/ollama.nix
+++ b/services/ollama.nix
@@ -1,6 +1,6 @@
 { config, pkgs, lib, ... }:
 {
-	config = lib.mkIf (config.networking.hostName == "FredOS-Mediaserver") {
+    config = lib.mkIf (config.networking.hostName == "FredOS-Mediaserver") {
 
         services.ollama.enable = true;
         services.ollama.acceleration = "cuda";
@@ -8,12 +8,17 @@
         services.open-webui.enable = true;
         services.open-webui.port = 8888;
 
-        #testing
         nixpkgs.overlays = [
             (final: prev: {
-                valkey = prev.valkey.overrideAttrs (oldAttrs: {
-                    doCheck = false; 
-                });
+                # We need to reach into the python package set
+                pythonPackagesExtensions = prev.pythonPackagesExtensions ++ [
+                    (python-final: python-prev: {
+                        langchain = python-prev.langchain.overridePythonAttrs (oldAttrs: {
+                            # This disables the checkPhase (running tests) for langchain
+                            doCheck = false; 
+                        });
+                    })
+                ];
             })
         ];
     };