diff --git a/netbox/users/api/views.py b/netbox/users/api/views.py
index 04b3ae336..8be1f85a2 100644
--- a/netbox/users/api/views.py
+++ b/netbox/users/api/views.py
@@ -70,7 +70,13 @@ class TokenProvisionView(APIView):
     """
     permission_classes = []
 
-    # @extend_schema(methods=["post"], responses={201: serializers.TokenSerializer})
+    @extend_schema(
+        request=serializers.TokenProvisionSerializer,
+        responses={
+            201: serializers.TokenSerializer,
+            401: OpenApiTypes.OBJECT,
+        }
+    )
     def post(self, request):
         serializer = serializers.TokenProvisionSerializer(data=request.data)
         serializer.is_valid()