Fix AVX bug