diff --git a/noiseutil/cipher_state_test.go b/noiseutil/cipher_state_test.go index cba2b47c..d93f2b12 100644 --- a/noiseutil/cipher_state_test.go +++ b/noiseutil/cipher_state_test.go @@ -22,7 +22,7 @@ func TestNewCipherStateDispatch(t *testing.T) { encA, _ := buildCipherStates(t, CipherAESGCM) encC, _ := buildCipherStates(t, noise.CipherChaChaPoly) - if CipherAESGCM == noise.CipherAESGCM { + if !EncryptLockNeeded { assert.IsType(t, &CipherStateAESGCM{}, NewCipherState(encA, CipherAESGCM)) } else { // fips140