Fix failing code