diff --git a/nix/module.nix b/nix/module.nix
index e6b8357..456ee58 100644
--- a/nix/module.nix
+++ b/nix/module.nix
@@ -32,8 +32,7 @@ in {
       description = "Headscale Web UI";
 
       wantedBy = ["multi-user.target"];
-      # TODO: Integrate with `headscale` service.
-      after = ["network.target"];
+      after = ["network.target" "headscale.service"];
 
       environment = mapAttrs (_: toString) cfg.settings;