[LIT] Fix failing LIT tests