Fix typo in computation.py