Removed unused import from test_demo.py