Make 'simple' example work again

I think this is self-explaning