fix: clear wg after handle command

command.go

@@ -72,10 +72,9 @@ func (srv *Server) consumeCommands(ctx context.Context, conn net.Conn, reader *b
 		// NOTE: we increase the wait group by one in order to make sure that idle
 		// connections are not blocking a close.
 		srv.wg.Add(1)
-		defer srv.wg.Done()
-
 		srv.logger.Debug("incoming command", slog.Int("length", length), slog.String("type", string(t)))
 		err = srv.handleCommand(ctx, conn, t, reader, writer)
+		srv.wg.Done()
 		if errors.Is(err, io.EOF) {
 			return nil
 		}