fixing test_compute.py