diff --git a/scripts/solve_network.py b/scripts/solve_network.py
index 4b988666..203d8b0f 100644
--- a/scripts/solve_network.py
+++ b/scripts/solve_network.py
@@ -835,7 +835,8 @@ def extra_functionality(n, snapshots):
         sys.path.append(os.path.dirname(source_path))
         module_name = os.path.splitext(os.path.basename(source_path))[0]
         module = importlib.import_module(module_name)
-        module.custom_extra_functionality(n, snapshots, snakemake)
+        custom_extra_functionality = getattr(module, module_name)
+        custom_extra_functionality(n, snapshots, snakemake)
 
 
 def solve_network(n, config, solving, opts="", **kwargs):