From c5734183e9ff8124447142c646bf4d4c690e2ab4 Mon Sep 17 00:00:00 2001
From: MartinGC94 <MGC94@outlook.dk>
Date: Sun, 29 Sep 2024 21:44:11 +0200
Subject: [PATCH] Fix test

---
 .../Language/SemanticTokenTest.cs                           | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/test/PowerShellEditorServices.Test/Language/SemanticTokenTest.cs b/test/PowerShellEditorServices.Test/Language/SemanticTokenTest.cs
index c343363429..196fa9ca6d 100644
--- a/test/PowerShellEditorServices.Test/Language/SemanticTokenTest.cs
+++ b/test/PowerShellEditorServices.Test/Language/SemanticTokenTest.cs
@@ -114,7 +114,11 @@ function Get-A*A {
                         Assert.Single(mappedTokens, sToken => SemanticTokenType.Keyword == sToken.Type);
                         break;
                     case "Get-A*A":
-                        Assert.Single(mappedTokens, sToken => SemanticTokenType.Function == sToken.Type);
+                        if (t.TokenFlags.HasFlag(TokenFlags.CommandName))
+                        {
+                            Assert.Single(mappedTokens, sToken => SemanticTokenType.Function == sToken.Type);
+                        }
+
                         break;
                 }
             }