diff --git a/extension.neon b/extension.neon
index 63673855..e212fe72 100644
--- a/extension.neon
+++ b/extension.neon
@@ -18,6 +18,7 @@ parameters:
 			- Doctrine\ORM\Query
 	stubFiles:
 		- stubs/Criteria.stub
+		- stubs/DBAL/Exception/UniqueConstraintViolationException.stub
 		- stubs/DocumentManager.stub
 		- stubs/DocumentRepository.stub
 		- stubs/EntityManager.stub
@@ -38,6 +39,7 @@ parameters:
 		- stubs/ORM/AbstractQuery.stub
 		- stubs/ORM/Mapping/ClassMetadata.stub
 		- stubs/ORM/Mapping/ClassMetadataInfo.stub
+		- stubs/ORM/ORMException.stub
 		- stubs/ORM/Query.stub
 		- stubs/Persistence/Mapping/ClassMetadata.stub
 		- stubs/ServiceDocumentRepository.stub
diff --git a/stubs/DBAL/Exception/UniqueConstraintViolationException.stub b/stubs/DBAL/Exception/UniqueConstraintViolationException.stub
new file mode 100644
index 00000000..aae74fc8
--- /dev/null
+++ b/stubs/DBAL/Exception/UniqueConstraintViolationException.stub
@@ -0,0 +1,8 @@
+<?php
+
+namespace Doctrine\DBAL\Exception;
+
+class UniqueConstraintViolationException extends \Exception
+{
+
+}
diff --git a/stubs/EntityManager.stub b/stubs/EntityManager.stub
index ee2c5c17..6b703d53 100644
--- a/stubs/EntityManager.stub
+++ b/stubs/EntityManager.stub
@@ -56,4 +56,13 @@ class EntityManager implements EntityManagerInterface
 	 */
 	public function copy($entity, $deep = false);
 
+	/**
+     * @param object|object[]|null $entity
+     * @return void
+     *
+     * @throws ORMException
+     * @throws \Doctrine\DBAL\Exception\UniqueConstraintViolationException
+     */
+    public function flush($entity = null);
+
 }
diff --git a/stubs/ORM/ORMException.stub b/stubs/ORM/ORMException.stub
new file mode 100644
index 00000000..86d6acb8
--- /dev/null
+++ b/stubs/ORM/ORMException.stub
@@ -0,0 +1,8 @@
+<?php
+
+namespace Doctrine\ORM;
+
+class ORMException extends \Exception
+{
+
+}