Restore imports

1 files changed, 2 insertions, 0 deletions

diff --git a/install.py b/install.py
index 2a52159..e2c5306 100755
--- a/install.py
+++ b/install.py
@@ -55,6 +55,8 @@ loc['address']  = raw_input('4/6 Address [Somestreet 1]: ') or 'Somestreet 1'
 loc['zip_code'] = raw_input('5/6 Postal code [1234]: ') or '1234'
 loc['city']     = raw_input('6/6 City [Stockholm]: ') or 'Stockholm'
 
+from servo.models import Location, User
+
 first_loc = Location(**loc)
 first_loc.save()
 su = User.objects.filter(pk=1).update(location=first_loc)