From 16fcbd1824727697e8a6625ae56ab7b55374d4ca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alvaro=20Mun=CC=83oz?= <alvaro@pwntester.com>
Date: Wed, 11 Nov 2020 23:30:33 +0100
Subject: [PATCH] More improvements

---
 queries/ql/highlights.scm | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/queries/ql/highlights.scm b/queries/ql/highlights.scm
index 95be42ccc..fde811af6 100644
--- a/queries/ql/highlights.scm
+++ b/queries/ql/highlights.scm
@@ -109,6 +109,8 @@
 (datatype name: (className) @type)
 (typeExpr name: (className) @type)
 
+(importModuleExpr name: (simpleId) @variable)
+(qualModuleExpr name: (simpleId) @variable)
 (varName) @variable
 
 (integer) @number