Add shape mismatch test for corr