diff --git a/scripts/base_network.py b/scripts/base_network.py
index 346f99a59..77d3d2507 100644
--- a/scripts/base_network.py
+++ b/scripts/base_network.py
@@ -698,6 +698,16 @@ def _adjust_capacities_of_under_construction_branches(n, config):
     return n
 
 
+def _set_shapes(n, country_shapes, offshore_shapes):
+    # Write the geodataframes country_shapes and offshore_shapes to the network.shapes component
+    country_shapes = gpd.read_file(country_shapes).rename(columns={"name": "idx"})
+    country_shapes["type"] = "country"
+    offshore_shapes = gpd.read_file(offshore_shapes).rename(columns={"name": "idx"})
+    offshore_shapes["type"] = "offshore"
+    all_shapes = pd.concat([country_shapes, offshore_shapes])
+    n.shapes = pd.concat([n.shapes, all_shapes])
+
+
 def base_network(
     eg_buses,
     eg_converters,
@@ -758,13 +768,19 @@ def base_network(
 
     n = _adjust_capacities_of_under_construction_branches(n, config)
 
+    _set_shapes(n, country_shapes, offshore_shapes)
+
     return n
 
 
 if __name__ == "__main__":
     if "snakemake" not in globals():
+        import os
+
         from _helpers import mock_snakemake
 
+        os.chdir(os.path.dirname(os.path.abspath(__file__)))
+
         snakemake = mock_snakemake("base_network")
     configure_logging(snakemake)
     set_scenario_config(snakemake)