diff --git a/blake2b/blake2b_test.go b/blake2b/blake2b_test.go
index a38fceb..7954346 100644
--- a/blake2b/blake2b_test.go
+++ b/blake2b/blake2b_test.go
@@ -22,7 +22,7 @@ func fromHex(s string) []byte {
 
 func TestHashes(t *testing.T) {
 	defer func(sse4, avx, avx2 bool) {
-		useSSE4, useAVX, useAVX2 = sse4, useAVX, avx2
+		useSSE4, useAVX, useAVX2 = sse4, avx, avx2
 	}(useSSE4, useAVX, useAVX2)
 
 	if useAVX2 {