Fix Python example