Partial Fix for python example