diff --git a/main.py b/main.py
index 435f5ac1155d0c418667e2d2e265e966c7de7acb..acbf1d1dc1f9574685946d77e320522f99107a86 100644
--- a/main.py
+++ b/main.py
@@ -125,7 +125,7 @@ print("\n--- Extension <-> Extension")
 
 should_look_for_connections = True
 while should_look_for_connections:
-    som_nfo["soma_w_ext_lmp"] = soma_t.SomasWithExtensionsLMap(somas)
+    som_nfo["soma_w_ext_lmp"] = soma_t.SomasLMap(somas)
     som_nfo["dist_to_closest"], som_nfo["influence_map"] = soma_t.InfluenceMaps(
         som_nfo["soma_w_ext_lmp"]
     )