Make the demo script work

Damn, I wanted to use it as an illustration of a problem I was having...