X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/60b06883b6d5a336ef47c01103ec1ce25aafae69..2d8bcd45b873aa01d43b07d0179d373e0bdc9bb5:/apps/catalogue/utils.py?ds=sidebyside

diff --git a/apps/catalogue/utils.py b/apps/catalogue/utils.py
index 02e5b6d93..368f96d20 100644
--- a/apps/catalogue/utils.py
+++ b/apps/catalogue/utils.py
@@ -49,13 +49,13 @@ class ExistingFile(UploadedFile):
 
 
 class BookImportDocProvider(DocProvider):
-    """ used for joined EPUBs """
+    """Used for joined EPUB and PDF files."""
 
     def __init__(self, book):
         self.book = book
-        
+
     def by_slug(self, slug):
         if slug == self.book.slug:
             return self.book.xml_file
         else:
-            return Book.objects.get(slug=slug).xml_file
+            return type(self.book).objects.get(slug=slug).xml_file