diff --git a/dj_rest_auth/tests/test_api.py b/dj_rest_auth/tests/test_api.py
index 4c4de9e..c53be66 100644
--- a/dj_rest_auth/tests/test_api.py
+++ b/dj_rest_auth/tests/test_api.py
@@ -583,13 +583,18 @@ class APIBasicTests(TestsMixin, TestCase):
         get_user_model().objects.create_user(self.USERNAME, '', self.PASS)
         resp = self.post(self.login_url, data=payload, status_code=200)
         token = resp.data['refresh_token']
+        # test refresh token not included in request data
         resp = self.post(self.logout_url, status=200)
         self.assertEqual(resp.status_code, 401)
+        # test token is invalid or expired
         resp = self.post(self.logout_url, status=200, data={'refresh': '1'})
-        self.assertEqual(resp.status_code, 404)
+        self.assertEqual(resp.status_code, 401)
+        # test successful logout
         resp = self.post(self.logout_url, status=200, data={'refresh': token})
         self.assertEqual(resp.status_code, 200)
+        # test token is blacklisted
         resp = self.post(self.logout_url, status=200, data={'refresh': token})
-        self.assertEqual(resp.status_code, 404)
+        self.assertEqual(resp.status_code, 401)
+        # test other TokenError, AttributeError, TypeError (invalid format)
         resp = self.post(self.logout_url, status=200, data=json.dumps({'refresh': token}))
         self.assertEqual(resp.status_code, 500)
diff --git a/dj_rest_auth/views.py b/dj_rest_auth/views.py
index 325466e..25ebe31 100644
--- a/dj_rest_auth/views.py
+++ b/dj_rest_auth/views.py
@@ -159,7 +159,7 @@ class LogoutView(APIView):
                     if hasattr(error, 'args'):
                         if 'Token is blacklisted' in error.args or 'Token is invalid or expired' in error.args:
                             response = Response({"detail": _(error.args[0])},
-                                                status=status.HTTP_404_NOT_FOUND)
+                                                status=status.HTTP_401_UNAUTHORIZED)
 
                         else:
                             response = Response({"detail": _("An error has occurred.")},