Fix module path

Fix module path in `main.py`.