From 006a86d9191276221c4d8df19fd0745560b66227 Mon Sep 17 00:00:00 2001
From: josh <josh@elixxir.io>
Date: Tue, 9 Mar 2021 12:01:27 -0800
Subject: [PATCH] Fix refactor of storage KV

---
 storage/clientVersion/store.go      | 4 ++--
 storage/clientVersion/store_test.go | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/storage/clientVersion/store.go b/storage/clientVersion/store.go
index 344b138bd..3f1eca48a 100644
--- a/storage/clientVersion/store.go
+++ b/storage/clientVersion/store.go
@@ -45,7 +45,7 @@ func LoadStore(kv *versioned.KV) (*Store, error) {
 		kv: kv.Prefix(prefix),
 	}
 
-	obj, err := s.kv.Get(storeKey)
+	obj, err := s.kv.Get(storeKey, storeVersion)
 	if err != nil {
 		return nil, err
 	}
@@ -113,5 +113,5 @@ func (s *Store) save() error {
 		Data:      []byte(s.version.String()),
 	}
 
-	return s.kv.Set(storeKey, &obj)
+	return s.kv.Set(storeKey, storeVersion, &obj)
 }
diff --git a/storage/clientVersion/store_test.go b/storage/clientVersion/store_test.go
index 8ff38a275..22da1257c 100644
--- a/storage/clientVersion/store_test.go
+++ b/storage/clientVersion/store_test.go
@@ -71,7 +71,7 @@ func TestLoadStore_ParseVersionError(t *testing.T) {
 		Data:      []byte("invalid version"),
 	}
 
-	err := kv.Prefix(prefix).Set(storeKey, &obj)
+	err := kv.Prefix(prefix).Set(storeKey, storeVersion, &obj)
 	if err != nil {
 		t.Fatalf("Failed to save Store: %+v", err)
 	}
@@ -219,7 +219,7 @@ func TestStore_save(t *testing.T) {
 		t.Errorf("save() returned an error: %+v", err)
 	}
 
-	obj, err := s.kv.Get(storeKey)
+	obj, err := s.kv.Get(storeKey, storeVersion)
 	if err != nil {
 		t.Errorf("Failed to load clientVersion store: %+v", err)
 	}
-- 
GitLab