From 436a57234d24bb9b45c952c34e5e395d578a1dd5 Mon Sep 17 00:00:00 2001
From: WMJ <wmj@live.cn>
Date: Mon, 17 Apr 2023 21:04:28 +0800
Subject: [PATCH] - Fixed incorrectly implemented GetFollowingDirective

---
 Codist/Helpers/CodeAnalysisHelper.cs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Codist/Helpers/CodeAnalysisHelper.cs b/Codist/Helpers/CodeAnalysisHelper.cs
index 363155c1..30eb5082 100644
--- a/Codist/Helpers/CodeAnalysisHelper.cs
+++ b/Codist/Helpers/CodeAnalysisHelper.cs
@@ -970,13 +970,13 @@ public static TEndDirective GetFollowingDirective<TStartDirective, TEndDirective
 			DirectiveTriviaSyntax d = directive;
 			int c = 1;
 			while ((d = d.GetNextDirective()) != null) {
-				if (d.IsKind(SyntaxKind.EndRegionDirectiveTrivia)) {
+				if (d.IsKind(endSyntaxKind)) {
 					--c;
 					if (c == 0) {
 						return d as TEndDirective;
 					}
 				}
-				else if (d.IsKind(SyntaxKind.RegionDirectiveTrivia)) {
+				else if (d.IsKind(startSyntaxKind)) {
 					++c;
 				}
 			}