diff --git a/puml_generator.py b/puml_generator.py
index 9b1ef6a..93be27c 100644
--- a/puml_generator.py
+++ b/puml_generator.py
@@ -243,8 +243,9 @@ def class_to_puml(filename:str) -> JavaClass:
                     class_methods[method_name] = get_method_visibility(javaline)
                 next_line_setter = False
             elif javaline.startswith("import"):
-                importline_package_slug = javaline.strip().split(" ")[1].replace(";", "")
-                imported_classes[importline_package_slug.split(".")[-1]] = importline_package_slug
+                if "*" not in javaline:
+                    importline_package_slug = javaline.strip().split(" ")[1].replace(";", "")
+                    imported_classes[importline_package_slug] = importline_package_slug
 
             if javaline.startswith("@"):
                 if "@Getter" in javaline and "@Setter" in javaline: