Fixing a typo in unit-matrix