From bf0b2819e59a75fbb4120de408f8b2e916abe0cc Mon Sep 17 00:00:00 2001
From: Shunsuke Shibayama <sbym1346@gmail.com>
Date: Fri, 30 Sep 2022 02:21:48 +0900
Subject: [PATCH] Update server.rs

---
 src/server.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/server.rs b/src/server.rs
index c4b50c1..3c42b92 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -324,8 +324,8 @@ impl Server {
             let mut item = CompletionItem::new_simple(name.to_string(), vi.t.to_string());
             item.kind = match &vi.t {
                 Type::Subr(_) => Some(CompletionItemKind::FUNCTION),
-                Type::Module => Some(CompletionItemKind::MODULE),
                 Type::Class | Type::Trait => Some(CompletionItemKind::CLASS),
+                t if &t.name()[..] == "Module" || &t.name()[..] == "GenericModule" => Some(CompletionItemKind::MODULE),
                 _ if vi.muty.is_const() => Some(CompletionItemKind::CONSTANT),
                 _ => Some(CompletionItemKind::VARIABLE),
             };