Fixing python wrapper.