Call to undefined method setFileExtension for Doctrine YamlDriver

[ako977]

This is the full error I get:

Fatal error: Call to undefined method Doctrine\ORM\Mapping\Driver\YamlDriver::setFileExtension() in *\vendor\taluu\doctrine-orm-provider\lib\Nutwerk\Provider\DoctrineORMServiceProvider.php on line 104

I'm guessing the solution to be that on the line:

$driver = new YamlDriver((array)$entity['path']);
$driver->setFileExtension('.yml');

It should be instead:

$driver = new YamlDriver((array)$entity['path'],".yml");