Fix failing transformation test