diff --git a/modules/apps/swag/default.nix b/modules/apps/swag/default.nix
index c17a495..d524082 100644
--- a/modules/apps/swag/default.nix
+++ b/modules/apps/swag/default.nix
@@ -7,14 +7,14 @@ in {
   config = mkIf cfg.enable {
 
     networking.firewall = {
-      allowedTCPPorts = [ cfg.httpPort cfg.httpsPort ];
-      allowedUDPPorts = [ cfg.httpPort cfg.httpsPort ];
+      allowedTCPPorts = [ cfg.httpPort cfg.httpsPort 81 ];
+      allowedUDPPorts = [ cfg.httpPort cfg.httpsPort 81 ];
     };
 
     virtualisation.oci-containers.containers."${cfg.name}" = {
       autoStart = cfg.autoStart;
       image = cfg.image;
-      ports = [ "${toString cfg.httpPort}:80" "${toString cfg.httpsPort}:443" ];
+      ports = [ "${toString cfg.httpPort}:80" "${toString cfg.httpsPort}:443" "81:81" ];
       volumes = [ "${cfg.configPath}:/config" ];
       environment = {
         PUID = cfg.puid;
@@ -24,6 +24,7 @@ in {
         URL = cfg.url;
         VALIDATION = cfg.validation;
         SUBDOMAINS = cfg.subdomains;
+        DOCKER_MODS="linuxserver/mods:swag-dashboard";
       };
     };
   };