Loose of focus after cloasing Collection Editor

When user finishes with editing collections, using Collection Editor and push Close or Save or Cansel then Grid of Test scenario is loosing focus and user unable to navigate it by keyboard.

Suggested behaviour:
When Collection Editor disappears after editing Collection cell A then focus should be on the cell of A.