diff --git a/stubs/EntityManagerInterface.php b/stubs/EntityManagerInterface.php
index 5b7abf6..b1c47e4 100644
--- a/stubs/EntityManagerInterface.php
+++ b/stubs/EntityManagerInterface.php
@@ -29,7 +29,7 @@ public function find(string $entityName, $id, ?int $lockMode = null, ?int $lockV
      * @param class-string<T> $entityName
      * @param mixed           $id
      *
-     * @return T
+     * @return T|null
      *
      * @template T
      */
diff --git a/tests/acceptance/EntityManagerInterface.feature b/tests/acceptance/EntityManagerInterface.feature
index 86bdb99..dcb2020 100644
--- a/tests/acceptance/EntityManagerInterface.feature
+++ b/tests/acceptance/EntityManagerInterface.feature
@@ -90,7 +90,7 @@ Feature: EntityManagerInterface
     When I run Psalm
     Then I see these errors
       | Type            | Message                                      |
-      | InvalidArgument | Argument 1 of atan expects float, I provided |
+      | InvalidArgument | Argument 1 of atan expects float, I\|null provided |
     And I see no other errors
 
   @EntityManagerInterface::getReference