Fixing small problem with viewsets import