FIX: Keep all channels in cov after regularization