Fixing bug in new python wrapper.