diff --git a/psalm.xml b/psalm.xml
index eac4c173..ebb13347 100644
--- a/psalm.xml
+++ b/psalm.xml
@@ -25,6 +25,7 @@
 		<UndefinedClass>
 			<errorLevel type="suppress">
 				<referencedClass name="OC" />
+				<referencedClass name="OC\DB\Connection" />
 			</errorLevel>
 		</UndefinedClass>
 		<UndefinedDocblockClass>
@@ -33,6 +34,7 @@
 				<referencedClass name="Doctrine\DBAL\Schema\SchemaException" />
 				<referencedClass name="Doctrine\DBAL\Driver\Statement" />
 				<referencedClass name="Doctrine\DBAL\Schema\Table" />
+				<referencedClass name="OC\DB\Connection" />
 			</errorLevel>
 		</UndefinedDocblockClass>
 	</issueHandlers>
diff --git a/tests/psalm-baseline.xml b/tests/psalm-baseline.xml
index 993ea03b..b8182253 100644
--- a/tests/psalm-baseline.xml
+++ b/tests/psalm-baseline.xml
@@ -1,22 +1,10 @@
 <?xml version="1.0" encoding="UTF-8"?>
-<files psalm-version="4.30.0@d0bc6e25d89f649e4f36a534f330f8bb4643dd69">
+<files psalm-version="4.26.0@6998fabb2bf528b65777bf9941920888d23c03ac">
   <file src="lib/AP.php">
     <InvalidScalarArgument occurrences="1">
       <code>$level</code>
     </InvalidScalarArgument>
   </file>
-  <file src="lib/Migration/Version0003Date20200823023900.php">
-    <UndefinedDocblockClass occurrences="1">
-      <code>DBALException</code>
-    </UndefinedDocblockClass>
-  </file>
-  <file src="lib/Migration/Version0003Date20200823023911.php">
-    <UndefinedClass occurrences="3">
-      <code>SchemaException</code>
-      <code>SchemaException</code>
-      <code>Type</code>
-    </UndefinedClass>
-  </file>
   <file src="lib/Model/ActivityPub/ACore.php">
     <InvalidArgument occurrences="1">
       <code>['a', 'p', 'span', 'br']</code>