diff --git a/lib/Db/Token.php b/lib/Db/Token.php
index c5cfc91..8a9f4f5 100644
--- a/lib/Db/Token.php
+++ b/lib/Db/Token.php
@@ -9,6 +9,7 @@
 namespace OCA\TwoFactorNextcloudNotification\Db;
 
 use OCP\AppFramework\Db\Entity;
+use OCP\DB\Types;
 
 /**
  * @method string getUserId()
@@ -35,9 +36,9 @@ class Token extends Entity {
 	protected $timestamp;
 
 	public function __construct() {
-		$this->addType('userId', 'string');
-		$this->addType('token', 'string');
-		$this->addType('status', 'int');
-		$this->addType('timestamp', 'int');
+		$this->addType('userId', Types::STRING);
+		$this->addType('token', Types::STRING);
+		$this->addType('status', Types::INTEGER);
+		$this->addType('timestamp', Types::INTEGER);
 	}
 }