Fixed tests/test_main.py to pass CI build.