diff --git a/flake.lock b/flake.lock
index 61d58f4..22ee8cb 100644
--- a/flake.lock
+++ b/flake.lock
@@ -420,16 +420,15 @@
         "poetry2nixi": "poetry2nixi"
       },
       "locked": {
-        "lastModified": 1712437610,
-        "narHash": "sha256-U0xIqSupo6yBVWQcHIcLPDKg5ShQIkV/0Ay+Xg3GIlk=",
-        "ref": "tlater/implement-nix-module",
-        "rev": "43c4f591328f194ca989b86cea49773104f75549",
-        "revCount": 7,
+        "lastModified": 1733295746,
+        "narHash": "sha256-044WeOcY609lzNNAUdXfkVnyx3F2ImwTltDp1rMGH54=",
+        "ref": "refs/heads/main",
+        "rev": "3ebb230e05a7fff4e2d1f81079fe25b0334daad6",
+        "revCount": 12,
         "type": "git",
         "url": "ssh://git@github.com/sonnenshift/battery-manager"
       },
       "original": {
-        "ref": "tlater/implement-nix-module",
         "type": "git",
         "url": "ssh://git@github.com/sonnenshift/battery-manager"
       }
diff --git a/flake.nix b/flake.nix
index 6dbbaa5..ea27f2d 100644
--- a/flake.nix
+++ b/flake.nix
@@ -27,7 +27,7 @@
     };
 
     sonnenshift = {
-      url = "git+ssh://git@github.com/sonnenshift/battery-manager?ref=tlater/implement-nix-module";
+      url = "git+ssh://git@github.com/sonnenshift/battery-manager";
       inputs.nixpkgs.follows = "nixpkgs";
     };
   };