Add __init__.py to test