diff --git a/db/notemapper.php b/db/notemapper.php
index 50ce0c0..40f3dd7 100644
--- a/db/notemapper.php
+++ b/db/notemapper.php
@@ -1,13 +1,13 @@
 <?php
 namespace OCA\QuickNotes\Db;
 
-use OCP\IDb;
+use OCP\IDBConnection;
 use OCP\AppFramework\Db\Mapper;
 use OCP\AppFramework\Db\DoesNotExistException;
 
 class NoteMapper extends Mapper {
 
-	public function __construct(IDb $db) {
+	public function __construct(IDBConnection $db) {
 		parent::__construct($db, 'quicknotes_notes', '\OCA\QuickNotes\Db\Note');
 	}