BUG: fix missing .shape in sandwich_covariance refactoring