diff --git a/flake.lock b/flake.lock
index b6cca1d..2976eb2 100644
--- a/flake.lock
+++ b/flake.lock
@@ -25,7 +25,9 @@
         "flake-utils": [
           "flake-utils"
         ],
-        "nixpkgs": "nixpkgs_2",
+        "nixpkgs": [
+          "nixpkgs"
+        ],
         "scripts": "scripts"
       },
       "locked": {
@@ -96,7 +98,9 @@
         "flake-utils": [
           "flake-utils"
         ],
-        "nixpkgs": "nixpkgs_3",
+        "nixpkgs": [
+          "nixpkgs"
+        ],
         "scripts": "scripts_2"
       },
       "locked": {
@@ -110,20 +114,27 @@
         "type": "path"
       }
     },
-    "katex-source": {
-      "flake": false,
+    "katex": {
+      "inputs": {
+        "flake-utils": [
+          "flake-utils"
+        ],
+        "nixpkgs": [
+          "nixpkgs"
+        ]
+      },
       "locked": {
-        "lastModified": 1691244098,
-        "narHash": "sha256-hDHo7JQAo+fGxQvY5OtXlfh+e6PjlVIQPTyCa3Fjg0Y=",
-        "owner": "KaTeX",
+        "lastModified": 1696151934,
+        "narHash": "sha256-8kihcqdgYjoVuGozfgfcWh81yqMUvns4+C/fgkn+RNQ=",
+        "owner": "Smaug123",
         "repo": "KaTeX",
-        "rev": "4f1d9166749ca4bd669381b84b45589f1500a476",
+        "rev": "ac1f9b30441f63ea20216a36ffa7148dc0e9a9b3",
         "type": "github"
       },
       "original": {
-        "owner": "KaTeX",
+        "owner": "Smaug123",
+        "ref": "nix",
         "repo": "KaTeX",
-        "rev": "4f1d9166749ca4bd669381b84b45589f1500a476",
         "type": "github"
       }
     },
@@ -158,60 +169,14 @@
         "type": "github"
       }
     },
-    "nixpkgs_3": {
-      "locked": {
-        "lastModified": 1688392541,
-        "narHash": "sha256-lHrKvEkCPTUO+7tPfjIcb7Trk6k31rz18vkyqmkeJfY=",
-        "owner": "NixOS",
-        "repo": "nixpkgs",
-        "rev": "ea4c80b39be4c09702b0cb3b42eab59e2ba4f24b",
-        "type": "github"
-      },
-      "original": {
-        "owner": "NixOS",
-        "ref": "nixos-22.11",
-        "repo": "nixpkgs",
-        "type": "github"
-      }
-    },
-    "nixpkgs_4": {
-      "locked": {
-        "lastModified": 1688392541,
-        "narHash": "sha256-lHrKvEkCPTUO+7tPfjIcb7Trk6k31rz18vkyqmkeJfY=",
-        "owner": "NixOS",
-        "repo": "nixpkgs",
-        "rev": "ea4c80b39be4c09702b0cb3b42eab59e2ba4f24b",
-        "type": "github"
-      },
-      "original": {
-        "owner": "NixOS",
-        "ref": "nixos-22.11",
-        "repo": "nixpkgs",
-        "type": "github"
-      }
-    },
-    "nixpkgs_5": {
-      "locked": {
-        "lastModified": 1688392541,
-        "narHash": "sha256-lHrKvEkCPTUO+7tPfjIcb7Trk6k31rz18vkyqmkeJfY=",
-        "owner": "NixOS",
-        "repo": "nixpkgs",
-        "rev": "ea4c80b39be4c09702b0cb3b42eab59e2ba4f24b",
-        "type": "github"
-      },
-      "original": {
-        "owner": "NixOS",
-        "ref": "nixos-22.11",
-        "repo": "nixpkgs",
-        "type": "github"
-      }
-    },
     "pdfs": {
       "inputs": {
         "flake-utils": [
           "flake-utils"
         ],
-        "nixpkgs": "nixpkgs_5",
+        "nixpkgs": [
+          "nixpkgs"
+        ],
         "scripts": "scripts_3"
       },
       "locked": {
@@ -234,8 +199,8 @@
         "extra-content": "extra-content",
         "flake-utils": "flake-utils_2",
         "images": "images",
-        "katex-source": "katex-source",
-        "nixpkgs": "nixpkgs_4",
+        "katex": "katex",
+        "nixpkgs": "nixpkgs_2",
         "pdfs": "pdfs",
         "scripts": "scripts_4"
       }
diff --git a/flake.nix b/flake.nix
index d1c2b1e..3b2f449 100644
--- a/flake.nix
+++ b/flake.nix
@@ -9,20 +9,24 @@
       url = "path:/Users/patrick/Desktop/website/extra-site-content";
       flake = false;
     };
-    katex-source = {
-      url = "github:KaTeX/KaTeX/4f1d9166749ca4bd669381b84b45589f1500a476";
-      flake = false;
+    katex = {
+      url = "github:Smaug123/KaTeX/nix";
+      inputs.nixpkgs.follows = "nixpkgs";
+      inputs.flake-utils.follows = "flake-utils";
     };
     images = {
       url = "path:/Users/patrick/Desktop/website/static-site-images";
+      inputs.nixpkgs.follows = "nixpkgs";
       inputs.flake-utils.follows = "flake-utils";
     };
     pdfs = {
       url = "github:Smaug123/static-site-pdfs";
+      inputs.nixpkgs.follows = "nixpkgs";
       inputs.flake-utils.follows = "flake-utils";
     };
     anki-decks = {
       url = "github:Smaug123/anki-decks";
+      inputs.nixpkgs.follows = "nixpkgs";
       inputs.flake-utils.follows = "flake-utils";
     };
   };
@@ -31,7 +35,7 @@
     self,
     nixpkgs,
     flake-utils,
-    katex-source,
+    katex,
     images,
     pdfs,
     anki-decks,
@@ -48,36 +52,16 @@
       let
         buildHugo = scripts.lib.createShellScript pkgs "hugo" ./docker/hugo/build.sh;
       in let
-        katex = pkgs.stdenv.mkDerivation {
+        katex-parts = pkgs.stdenv.mkDerivation {
           __contentAddressed = true;
           pname = "katex";
           version = "0.1.0";
-          src = katex-source;
-
-          buildInputs = [pkgs.nodejs pkgs.yarn];
-
-          buildPhase = ''
-            export HOME=$(mktemp -d)
-            yarn --immutable
-            yarn build
-          '';
+          src = katex.outputs.packages.${system}.default;
 
           installPhase = ''
-            mkdir -p "$out/fonts"
-            cp ./fonts/* "$out/fonts"
-            cp -r ./dist "$out/dist"
-          '';
-        };
-      in let
-        extraContent = pkgs.stdenv.mkDerivation {
-          __contentAddressed = true;
-          pname = "patrickstevens.co.uk-extraContent";
-          version = "0.1.0";
-          src = extra-content;
-          buildInputs = [];
-          installPhase = ''
-            mkdir -p $out
-            cp -r ./. $out
+            mkdir "$out"
+            ls -la .
+            cp -r ./libexec/katex/dist "$out/dist"
           '';
         };
       in
@@ -94,7 +78,7 @@
           ];
 
           buildPhase = ''
-            ${scripts.lib.createShellScript pkgs "all" ./build/all.sh}/run.sh "${pdfs.packages.${system}.default}" "${images}" "${anki-decks.packages.${system}.default}" "${buildHugo}" "${katex}" "${extraContent}"
+            ${scripts.lib.createShellScript pkgs "all" ./build/all.sh}/run.sh "${pdfs.packages.${system}.default}" "${images}" "${anki-decks.packages.${system}.default}" "${buildHugo}" "${katex-parts}" "${extra-content}"
           '';
 
           checkPhase = ''