diff --git a/extract-passwords-for-email.py b/extract-passwords-for-email.py
index d72705792a7691b2b26c863841e6c4bb01efeafc..7769d02557a621138e724e74d2a6200bfd5dea55 100644
--- a/extract-passwords-for-email.py
+++ b/extract-passwords-for-email.py
@@ -5,12 +5,16 @@
 # python extract-passwords-for-email.py <election_uuid> <email_address>
 #
 
-from django.core.management import setup_environ
-import settings, sys, csv
+import sys
 
-setup_environ(settings)
+import csv
+import django
+import os
 
-from helios.models import *
+from helios.models import Election
+
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "settings")
+django.setup()
 
 election_uuid = sys.argv[1]
 email = sys.argv[2]