Use `k` to be consistent with the paper.