LPS-51188 Adds missing imports