LPS-47682 Missing import during the refactor