Merge c30242c into e41723b

burke-software · Jul 9, 2017 · 1775cfa · 1775cfa
2 parents e41723b + c30242c
commit 1775cfa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/report_builder/api/views.py b/report_builder/api/views.py
@@ -7,7 +7,7 @@
 from rest_framework.views import APIView
 from rest_framework.response import Response
 from rest_framework.permissions import IsAdminUser
-from ..models import Report, Format, FilterField
+from ..models import Report, Format, FilterField, get_allowed_models
 from .serializers import (
     ReportNestedSerializer, ReportSerializer, FormatSerializer,
     FilterFieldSerializer, ContentTypeSerializer)
@@ -44,7 +44,7 @@ class ContentTypeViewSet(ReportBuilderViewMixin, viewsets.ReadOnlyModelViewSet):
     """ Read only view of content types.
     Used to populate choices for new report root model.
     """
-    queryset = ContentType.objects.all()
+    queryset = get_allowed_models()
     serializer_class = ContentTypeSerializer
     permission_classes = (IsAdminUser,)