diff --git a/modules/endlessh.nix b/modules/endlessh.nix
index 4cd95e4..9194792 100644
--- a/modules/endlessh.nix
+++ b/modules/endlessh.nix
@@ -3,13 +3,13 @@
   config = {
     services = {
       endlessh-go = lib.mkIf (!builtins.elem 22 config.services.openssh.ports) {
-        enable = true;
+        enable = lib.mkDefault true;
         port = 22;
       };
     };
 
-    networking.firewall = lib.mkIf config.services.openssh.enable {
-      allowedTCPPorts = config.services.openssh.ports ++ [ 22 ];
+    networking.firewall = lib.mkIf config.services.endlessh-go.enable {
+      allowedTCPPorts = 22;
     };
   };
 }