Fix AVX problem in some tests