diff --git a/src/SortableListener.php b/src/SortableListener.php
index 8f55065..b4b2001 100644
--- a/src/SortableListener.php
+++ b/src/SortableListener.php
@@ -8,13 +8,14 @@
 use Kdyby\Doctrine\EntityManager;
 use Kdyby\Doctrine\QueryBuilder;
 use Kdyby\Events\Subscriber;
-use Nette\Object;
+use Nette\SmartObject;
 
 /**
  * @author David Matejka
  */
-class SortableListener extends Object implements Subscriber
+class SortableListener implements Subscriber
 {
+    use SmartObject;
 
 	/** @var \ReflectionProperty[] */
 	private $reflCache = [];