1 files changed, 17 insertions, 14 deletions
diff --git a/tests/admin_widgets/widgetadmin.py b/tests/admin_widgets/widgetadmin.py
index 3dd9997598..a03e044b9d 100644
--- a/tests/admin_widgets/widgetadmin.py
+++ b/tests/admin_widgets/widgetadmin.py
@@ -1,6 +1,9 @@
 from django.contrib import admin
 
-from . import models
+from .models import (
+    Advisor, Album, Band, Bee, Car, CarTire, Event, Inventory, Member, Profile,
+    School, User,
+)
 
 
 class WidgetAdmin(admin.AdminSite):
@@ -15,7 +18,7 @@ class CarAdmin(admin.ModelAdmin):
 class CarTireAdmin(admin.ModelAdmin):
     def formfield_for_foreignkey(self, db_field, request, **kwargs):
         if db_field.name == "car":
-            kwargs["queryset"] = models.Car.objects.filter(owner=request.user)
+            kwargs["queryset"] = Car.objects.filter(owner=request.user)
             return db_field.formfield(**kwargs)
         return super(CarTireAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)
 
@@ -36,21 +39,21 @@ class SchoolAdmin(admin.ModelAdmin):
 
 site = WidgetAdmin(name='widget-admin')
 
-site.register(models.User)
-site.register(models.Car, CarAdmin)
-site.register(models.CarTire, CarTireAdmin)
+site.register(User)
+site.register(Car, CarAdmin)
+site.register(CarTire, CarTireAdmin)
 
-site.register(models.Member)
-site.register(models.Band)
-site.register(models.Event, EventAdmin)
-site.register(models.Album, AlbumAdmin)
+site.register(Member)
+site.register(Band)
+site.register(Event, EventAdmin)
+site.register(Album, AlbumAdmin)
 
-site.register(models.Inventory)
+site.register(Inventory)
 
-site.register(models.Bee)
+site.register(Bee)
 
-site.register(models.Advisor)
+site.register(Advisor)
 
-site.register(models.School, SchoolAdmin)
+site.register(School, SchoolAdmin)
 
-site.register(models.Profile)
+site.register(Profile)