fix for my simple example

I don;t know if this is the best way