Fixed shape mismatch error test.