small fix
fix python example in light of @hectcastro comments