diff --git a/bundles/systemd-swap/metadata.py b/bundles/systemd-swap/metadata.py
index d4f6d8c..6b7dcce 100644
--- a/bundles/systemd-swap/metadata.py
+++ b/bundles/systemd-swap/metadata.py
@@ -3,6 +3,11 @@ defaults = {
     'systemd': {
         'units': {
             'swapfile.swap': {
+                'Unit': {
+                    'PartOf': {
+                        'swapfile-prepare.service', # restart together
+                    },
+                },
                 'Swap': {
                     'What': '/swapfile',
                 },
@@ -31,6 +36,9 @@ def unit(metadata):
                         'Before': {
                             'swapfile.swap',
                         },
+                        'PartOf': {
+                            'swapfile.swap', # restart together
+                        },
                     },
                     'Service': {
                         'Type': 'oneshot',