diff --git a/rest_auth/serializers.py b/rest_auth/serializers.py
index c217f3f..41414d7 100644
--- a/rest_auth/serializers.py
+++ b/rest_auth/serializers.py
@@ -126,7 +126,7 @@ class KnoxTokenSerializer(serializers.ModelSerializer):
     """
 
     class Meta:
-        model = knox.models.AuthToken
+        model = TokenModel
         fields = ('token',)