fix sort_matrix.py