diff --git a/lib/Db/AttachMapper.php b/lib/Db/AttachMapper.php
index 42a6701..13aca2f 100644
--- a/lib/Db/AttachMapper.php
+++ b/lib/Db/AttachMapper.php
@@ -12,7 +12,7 @@ use OCA\QuickNotes\Db\Attach;
 class AttachMapper extends QBMapper {
 
 	public function __construct(IDBConnection $db) {
-		parent::__construct($db, 'quicknotes_attach');
+		parent::__construct($db, 'quicknotes_attach', Attach::class);
 	}
 
 	/**
diff --git a/lib/Db/NoteMapper.php b/lib/Db/NoteMapper.php
index f9e2caf..f8e7148 100644
--- a/lib/Db/NoteMapper.php
+++ b/lib/Db/NoteMapper.php
@@ -10,7 +10,7 @@ use OCP\DB\QueryBuilder\IQueryBuilder;
 class NoteMapper extends QBMapper {
 
 	public function __construct(IDBConnection $db) {
-		parent::__construct($db, 'quicknotes_notes', 'OCA\QuickNotes\Db\Note');
+		parent::__construct($db, 'quicknotes_notes', Note::class);
 	}
 
 	/**