@@ -1,7 +1,9 @@
from django.test import TestCase
-from django.contrib.auth.models import User
from oscar.apps.wishlists.models import WishList
+from oscar.core.compat import get_user_model
+
+User = get_user_model()
class TestAWishlist(TestCase):