diff --git a/simplesat/sat/policy/priority_queue_policy.py b/simplesat/sat/policy/priority_queue_policy.py
index 4e0ce47..9080f17 100644
--- a/simplesat/sat/policy/priority_queue_policy.py
+++ b/simplesat/sat/policy/priority_queue_policy.py
@@ -8,7 +8,7 @@
 from simplesat.utils import DefaultOrderedDict, toposort, transitive_neighbors
 from simplesat.priority_queue import PriorityQueue, GroupPrioritizer
 from .policy import IPolicy
-from .policy_logger import PolicyLogger
+from .policy_logger import LoggedPolicy
 
 
 class PriorityQueuePolicy(IPolicy):
@@ -218,7 +218,4 @@ def _update_cache_from_assignments(self, assignments):
         assert ours == theirs, msg.format(ours, theirs, has_new_keys)
 
 
-def LoggedPriorityQueuePolicty(pool, installed_repository, *args, **kwargs):
-    policy = PriorityQueuePolicy(pool, installed_repository, *args, **kwargs)
-    logger = PolicyLogger(policy, extra_args=args, extra_kwargs=kwargs)
-    return logger
+LoggedPriorityQueuePolicty = LoggedPolicy(PriorityQueuePolicy)