From 9492a2b461322e15d0dca4221bead5085ee79055 Mon Sep 17 00:00:00 2001
From: Yury Palyanitsa <159263676+yury-palyanitsa@users.noreply.github.com>
Date: Tue, 10 Dec 2024 12:13:54 +0200
Subject: [PATCH] Fix RDT group visitor (#56)

---
 rdt_visitor.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/rdt_visitor.go b/rdt_visitor.go
index 294c658..df8814e 100644
--- a/rdt_visitor.go
+++ b/rdt_visitor.go
@@ -145,7 +145,8 @@ func (visitor *RdtVisitor) VisitUnion(ctx *rdt.UnionContext, target *UnknownShap
 }
 
 func (visitor *RdtVisitor) VisitGroup(ctx *rdt.GroupContext, target *UnknownShape) (Shape, error) {
-	return visitor.Visit(ctx.GetChildren()[0].(antlr.ParseTree), target)
+	// First and last nodes are terminal nodes
+	return visitor.Visit(ctx.GetChildren()[1].(antlr.ParseTree), target)
 }
 
 func (visitor *RdtVisitor) VisitReference(ctx *rdt.ReferenceContext, target *UnknownShape) (Shape, error) {