Fix dimensio typo

This caused a panic with b matrices that are not square.