Fix main.py test coverage implementation for Python 3.10

Fix main.py test coverage implementation for Python 3.10 #157