Fix error when running python sample on codebox.io