Remove entity button not showing in Complex Inline Entity Form after upgrading to 3.x from 1.x

We need to handle 2 cases:

• When the count of type of entity bundles involved is more than 1, then current code will work, as count($handler->getReferenceableEntities(NULL, 'CONTAINS', 2)) will return 2
• When the count of type of entity bundles involved is equal to 1, we need to count the items inside bundle array key. Ex: in my case, the count should have been of count($handler->getReferenceableEntities(NULL, 'CONTAINS', 2)['gallery_item'])

There is some SSH error, I can't create MR.
Adding a patch instead for the fix.

An even simpler patch.