Wrong parameter in hook_entity_transform()

Problem/Motivation

I guess we have a wrong parameter type in:
function hook_entity_transform(array &$build, \Drupal\Core\Entity\EntityInterface $entity, \Drupal\Core\Entity\Display\EntityViewDisplayInterface $display, $transform_mode) {
transform_api.api.php

Proposed resolution

I guess it should be EntityTransformDisplay $display instead.

collection_entity_transform(): Argument #3 ($display) must be of type Drupal\Core\Entity\Display\EntityViewDisplayInterface, Drupal\transform_api\Entity\EntityTransformDisplay given