Fix avx group return type for single argument