try to fix regex but not entirely sure why it fixed it

1 files changed, 5 insertions, 1 deletions

diff --git a/parsers.py b/parsers.py
index 4924fbc..f3c66bd 100644
--- a/parsers.py
+++ b/parsers.py
@@ -50,8 +50,12 @@ def parse_regex(path):
             nfa.transitions[end1][EPSILON].add(start2)
             return start1, end2
         elif node[0] == "KLEENE":
-            start, end = visit(node[1])
+            start = nfa.add_state(nfa.count(), False, False)
+            end = nfa.add_state(nfa.count(), False, False)
+            start1, end1 = visit(node[1])
             nfa.transitions[start][EPSILON].add(end)
+            nfa.transitions[start][EPSILON].add(start1)
+            nfa.transitions[end1][EPSILON].add(end)
             nfa.transitions[end][EPSILON].add(start)
             return start, end
         elif node[0] == "OR":