Fix failing test - had switched it in demo