diff --git a/startup_scripts/340_power_feeds.py b/startup_scripts/340_power_feeds.py
index fc7ee38..8660da5 100644
--- a/startup_scripts/340_power_feeds.py
+++ b/startup_scripts/340_power_feeds.py
@@ -1,6 +1,6 @@
 import sys
 
-from dcim.models import Rack, PowerPanel, PowerFeed
+from dcim.models import Rack, PowerPanel, PowerFeed, Site
 from startup_script_utils import *
 from tenancy.models import Tenant
 
@@ -10,7 +10,8 @@ if power_feeds is None:
   sys.exit()
 
 required_assocs = {
-  'power_panel': (PowerPanel, 'name')
+  'power_panel': (PowerPanel, 'name'),
+  'site': (Site, 'name')
 }
 
 optional_assocs = {