Added missing type parameter in wrapper.py.