diff --git a/modules/services/docker-mailserver/default.nix b/modules/services/docker-mailserver/default.nix
index 0620b85..4501a61 100644
--- a/modules/services/docker-mailserver/default.nix
+++ b/modules/services/docker-mailserver/default.nix
@@ -69,6 +69,11 @@ in
     };
     services.az-docker.enable = true;
 
+    systemd.services."nixos-upgrade" = {
+      serviceConfig = {
+        ExecStartPost = "${pkgs.systemd}/bin/systemctl restart restart-mailserver.service";
+      };
+    };
     virtualisation.oci-containers = {
       backend = "docker";
       containers."mailserver" = {