diff --git a/polymorphic_tree/managers.py b/polymorphic_tree/managers.py index 13ac24b..69fef41 100644 --- a/polymorphic_tree/managers.py +++ b/polymorphic_tree/managers.py @@ -4,9 +4,12 @@ import django from mptt.managers import TreeManager from mptt.querysets import TreeQuerySet -from polymorphic.manager import PolymorphicManager from polymorphic.query import PolymorphicQuerySet +try: + from polymorphic.managers import PolymorphicManager +except ImportError: + from polymorphic.manager import PolymorphicManager class PolymorphicMPTTQuerySet(TreeQuerySet, PolymorphicQuerySet): """