Update server.rs

erg-lang · Sep 29, 2022 · bf0b281 · bf0b281
1 parent 91af9f3
commit bf0b281
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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),
             };