diff --git a/aikido_zen/sinks/builtins_import.py b/aikido_zen/sinks/builtins_import.py index a39486bc..ba646f29 100644 --- a/aikido_zen/sinks/builtins_import.py +++ b/aikido_zen/sinks/builtins_import.py @@ -14,10 +14,11 @@ def _import(func, instance, args, kwargs, return_value): return name = getattr(return_value, "__package__") - if not name or "." in name: - # Make sure the name exists and that it's not a submodule + if not name: + # Make sure the name exists return - if name == "importlib_metadata": + name = name.split(".")[0] # Remove submodules + if name == "importlib" or name == "importlib_metadata": # Avoid circular dependencies return