Fix python sample