diff --git a/lib/Db/StackMapper.php b/lib/Db/StackMapper.php
index 0d1283f67..854767c63 100644
--- a/lib/Db/StackMapper.php
+++ b/lib/Db/StackMapper.php
@@ -23,7 +23,6 @@
 
 namespace OCA\Deck\Db;
 
-use OCP\AppFramework\Db\Entity;
 use OCP\IDBConnection;
 
 
@@ -39,7 +38,7 @@ class StackMapper extends DeckMapper implements IPermissionMapper {
 
 	/**
 	 * @param $id
-	 * @return Entity if not found
+	 * @return \OCP\AppFramework\Db\Entity if not found
 	 */
     public function find($id) {
         $sql = 'SELECT * FROM `*PREFIX*deck_stacks` ' .
@@ -54,7 +53,7 @@ class StackMapper extends DeckMapper implements IPermissionMapper {
     }
     
 
-    public function delete(Entity $entity) {
+    public function delete(\OCP\AppFramework\Db\Entity $entity) {
         // delete cards on stack
 		$this->cardMapper->deleteByStack($entity->getId());
         return parent::delete($entity);