From 46f562e91d86e5aaf7c637926f4772acc982e57a Mon Sep 17 00:00:00 2001
From: mcrakhman <rakhmanov.m@gmail.com>
Date: Tue, 11 Oct 2022 11:15:46 +0200
Subject: [PATCH] Change mutex logic

---
 node/storage/treestorage.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/node/storage/treestorage.go b/node/storage/treestorage.go
index 68e88bfd..224cf292 100644
--- a/node/storage/treestorage.go
+++ b/node/storage/treestorage.go
@@ -123,11 +123,11 @@ func (t *treeStorage) Heads() ([]string, error) {
 }
 
 func (t *treeStorage) SetHeads(heads []string) (err error) {
+	t.headsMx.Lock()
+	defer t.headsMx.Unlock()
 	defer func() {
 		if err == nil {
-			t.headsMx.Lock()
 			t.heads = heads
-			t.headsMx.Unlock()
 		}
 	}()
 	payload := createHeadsPayload(heads)