Added missing import to testwrap.py