Don't cover __main__ section during tests