diff --git a/mace/tools/finetuning_utils.py b/mace/tools/finetuning_utils.py
index 71459fcd..71ca6a7c 100644
--- a/mace/tools/finetuning_utils.py
+++ b/mace/tools/finetuning_utils.py
@@ -103,7 +103,7 @@ def load_foundations_elements(
             )
         if (
             model.interactions[i].__class__.__name__
-            in ["RealAgnosticResidualInteractionBlock", "RealAgnosticDensityResidualInteractionBlock"]
+            in ["RealAgnosticDensityInteractionBlock", "RealAgnosticDensityResidualInteractionBlock"]
         ):
             # Assuming only 1 layer in density_fn
             getattr(model.interactions[i].density_fn, "layer0").weight = (