From 3050cc8659a5d2286cc38f00f4a0cd8f16e0e3f5 Mon Sep 17 00:00:00 2001
From: dovi <dovi.rivkin@gmail.com>
Date: Wed, 3 May 2023 14:42:31 -0400
Subject: [PATCH] fix WAIT spec & clean code

---
 packages/client/lib/commands/WAIT.spec.ts | 7 +++++--
 packages/client/lib/commands/WAIT.ts      | 2 +-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/packages/client/lib/commands/WAIT.spec.ts b/packages/client/lib/commands/WAIT.spec.ts
index 8a8fb3a36c..2b8c1de319 100644
--- a/packages/client/lib/commands/WAIT.spec.ts
+++ b/packages/client/lib/commands/WAIT.spec.ts
@@ -10,10 +10,13 @@ describe('WAIT', () => {
     );
   });
 
-  testUtils.testWithClient('client.wait', async client => {
+  testUtils.testAll('wait', async client => {
     assert.equal(
       await client.wait(0, 1),
       0
     );
-  }, GLOBAL.SERVERS.OPEN);
+  }, {
+    client: GLOBAL.SERVERS.OPEN,
+    cluster: GLOBAL.CLUSTERS.OPEN
+  });
 });
diff --git a/packages/client/lib/commands/WAIT.ts b/packages/client/lib/commands/WAIT.ts
index 5c4bd156a6..21c39a643e 100644
--- a/packages/client/lib/commands/WAIT.ts
+++ b/packages/client/lib/commands/WAIT.ts
@@ -1,4 +1,4 @@
-import { RedisArgument, SimpleStringReply, Command, NumberReply } from '../RESP/types';
+import { NumberReply, Command } from '../RESP/types';
 
 export default {
   FIRST_KEY_INDEX: undefined,