fixed typo in wrapper.py