LPS-46931 Fix import

cc @ealonso