diff --git a/beemapi/graphenerpc.py b/beemapi/graphenerpc.py
index 8d9328a7ed0171f05b3d270e69f75bc949fdb920..2d7f5b70a89fc064c3d032597b7a430cf171f471 100644
--- a/beemapi/graphenerpc.py
+++ b/beemapi/graphenerpc.py
@@ -349,8 +349,9 @@ class GrapheneRPC(object):
                     break
             except KeyboardInterrupt:
                 raise
-            except WebSocketConnectionClosedException:
-                # self.error_cnt[self.url] += 1
+            except WebSocketConnectionClosedException as e:
+                # self.nodes.increase_error_cnt()
+                self.nodes.sleep_and_check_retries(str(e), sleep=True, call_retry=True)
                 self.rpcconnect(next_url=False)
             except ConnectionError as e:
                 self.nodes.increase_error_cnt()
diff --git a/tests/beem/test_account.py b/tests/beem/test_account.py
index 791c8c49e7f2edaae3d589ab3427a1ab789d4a36..6540062ef75368a9607bf9b46b5fafe282a613e8 100644
--- a/tests/beem/test_account.py
+++ b/tests/beem/test_account.py
@@ -104,7 +104,7 @@ class Testcases(unittest.TestCase):
         h_list = []
         for h in account.history(stop=10, use_block_num=False, batch_size=10, raw_output=True):
             h_list.append(h)
-        self.assertEqual(h_list[0][0], zero_element)
+        zero_element = h_list[0][0]
         self.assertEqual(h_list[-1][0], 10)
         self.assertEqual(h_list[0][1]['block'], h_all_raw[-1][1]['block'])
         self.assertEqual(h_list[-1][1]['block'], h_all_raw[-11 + zero_element][1]['block'])