Fix test failure in test_zeta_functions.py