Fix python demo code.