diff --git a/src/symboltable.rs b/src/symboltable.rs index 04e5d9e..2e16fda 100644 --- a/src/symboltable.rs +++ b/src/symboltable.rs @@ -301,19 +301,17 @@ impl SymbolTableBuilder { if let Some(alias) = &part.alias { // `import mymodule as myalias` self.register_name(alias, SymbolRole::Assigned)?; + } else if part.symbols.is_empty() { + // `import module` + self.register_name(&part.module, SymbolRole::Assigned)?; } else { - if part.symbols.is_empty() { - // `import module` - self.register_name(&part.module, SymbolRole::Assigned)?; - } else { - // `from mymodule import myimport` - for symbol in &part.symbols { - if let Some(alias) = &symbol.alias { - // `from mymodule import myimportname as myalias` - self.register_name(alias, SymbolRole::Assigned)?; - } else { - self.register_name(&symbol.symbol, SymbolRole::Assigned)?; - } + // `from mymodule import myimport` + for symbol in &part.symbols { + if let Some(alias) = &symbol.alias { + // `from mymodule import myimportname as myalias` + self.register_name(alias, SymbolRole::Assigned)?; + } else { + self.register_name(&symbol.symbol, SymbolRole::Assigned)?; } } }