Fix pytest.