Fix python example so it runs