Bugfix: adjust size of identity matrix