Fixes test_linalg.py.