Fix broken demo script

- Move logging line to correct place