Rename module reference in main.py