diff --git a/__test__/genBlocksInterval.test.ts b/__test__/genBlocksInterval.test.ts
index 0c6c04c..cdc407b 100644
--- a/__test__/genBlocksInterval.test.ts
+++ b/__test__/genBlocksInterval.test.ts
@@ -13,11 +13,11 @@ beforeAll(async () => {
     devBlockIntervalMs: 100,
     tcpPort: udpAndTcpPort,
     udpPort: udpAndTcpPort,
-    jsonrpcHttpPort: jsonrpcHttpPort
+    jsonrpcHttpPort: jsonrpcHttpPort,
   });
 
   await server.start();
-  return  () => server.stop();
+  return () => server.stop();
 });
 
 describe("genBlocksInterval", () => {
diff --git a/__test__/server.test.ts b/__test__/server.test.ts
index c472734..dcd3aa2 100644
--- a/__test__/server.test.ts
+++ b/__test__/server.test.ts
@@ -23,8 +23,6 @@ describe("server", () => {
     expect(status.networkId).toBe(1234);
 
     await server.stop();
-    // TODO UPDATE
-    await wait(2000);
     await expect(async () => client.getStatus()).rejects.toThrow();
   });
 });