Removed unused import from example/test.py