Add more prints and the demo works, whatever...