also modify test error

pokt-network · Aug 22, 2024 · 508addc · 508addc
1 parent 39d738d
commit 508addc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kvstore/badger/kvstore_test.go b/kvstore/badger/kvstore_test.go
@@ -7,6 +7,7 @@ import (
 
 	"github.com/stretchr/testify/require"
 
+	badgerv4 "github.com/dgraph-io/badger/v4"
 	"github.com/pokt-network/smt/kvstore/badger"
 )
 
@@ -280,7 +281,7 @@ func TestBadger_KVStore_Exists(t *testing.T) {
 
 	// Key does not exist
 	exists, err = store.Exists([]byte("oof"))
-	require.ErrorIs(t, err, badger.ErrBadgerUnableToGetValue)
+	require.ErrorIs(t, err, badgerv4.ErrKeyNotFound)
 	require.False(t, exists)
 
 	err = store.Stop()