diff --git a/flake.nix b/flake.nix
index b7a2dc8..2706707 100644
--- a/flake.nix
+++ b/flake.nix
@@ -11,6 +11,7 @@
   outputs = { self, nixpkgs, crane, flake-utils }:
     let
       description = "blazingly-fast super-concurrent webscale file server";
+      src = ./.;
       packages = flake-utils.lib.eachDefaultSystem (system:
         let
           pkgs = nixpkgs.legacyPackages."${system}";
@@ -20,7 +21,7 @@
             pname = "u";
             version = "0.1.0";
             # Don't use cleanCargoSource here, as it removes sqlx-data.json
-            src = ./.;
+            src = src;
             doCheck = false;
           };
 
@@ -97,7 +98,7 @@
 
                 # Create and migrate the database
                 ${pkgs.sqlx-cli}/bin/sqlx database create
-                ${pkgs.sqlx-cli}/bin/sqlx migrate run
+                ${pkgs.sqlx-cli}/bin/sqlx migrate run --source "${src}/migrations"
               '';
 
               script = "${pkg}/bin/u";