Fixed IdentityMatrix creation