diff --git a/opt/dbs.py.src b/opt/dbs.py.src
index b7320b5..6d2fc5c 100644
--- a/opt/dbs.py.src
+++ b/opt/dbs.py.src
@@ -3,7 +3,7 @@
 # BEGIN HEROKU PYTHON/DJANGO LANGUAGE PACK
 # Dynamically set DATABASES from ENV vars
 
-import os, urlparse
+import os, re, urlparse
 urlparse.uses_netloc.append("postgres")
 key_re = re.compile(r"^(?P<name>[A-Z_]+)_URL$")
 
diff --git a/test/__main__.py b/test/__main__.py
index 91f82d3..d08d1f2 100644
--- a/test/__main__.py
+++ b/test/__main__.py
@@ -1,5 +1,4 @@
 import os
-import re
 import unittest
 
 BIN_DIR = os.path.dirname(__file__)