diff --git a/examples/cfd/external_aerodynamics/transformer_models/src/conf/data/volume.yaml b/examples/cfd/external_aerodynamics/transformer_models/src/conf/data/volume.yaml
index 247cbf4996..163bd40c17 100644
--- a/examples/cfd/external_aerodynamics/transformer_models/src/conf/data/volume.yaml
+++ b/examples/cfd/external_aerodynamics/transformer_models/src/conf/data/volume.yaml
@@ -27,4 +27,5 @@ data_keys:
   - "stl_faces"
   - "stl_centers"
   - "stl_coordinates"
-  
\ No newline at end of file
+  - "air_density"
+  - "stream_velocity"
diff --git a/examples/cfd/external_aerodynamics/transformer_models/src/conf/geotransolver_surface.yaml b/examples/cfd/external_aerodynamics/transformer_models/src/conf/geotransolver_surface.yaml
index ad58124f40..214e15bce7 100644
--- a/examples/cfd/external_aerodynamics/transformer_models/src/conf/geotransolver_surface.yaml
+++ b/examples/cfd/external_aerodynamics/transformer_models/src/conf/geotransolver_surface.yaml
@@ -39,7 +39,7 @@ data:
   include_sdf: false
   include_geometry: true
   geometry_sampling: 300_000
-  broadcast_global_features: true
+  broadcast_global_features: false
 
 
 # Logging configuration
diff --git a/examples/cfd/external_aerodynamics/transformer_models/src/inference_on_vtk.py b/examples/cfd/external_aerodynamics/transformer_models/src/inference_on_vtk.py
index 45505564e0..ae461fca02 100644
--- a/examples/cfd/external_aerodynamics/transformer_models/src/inference_on_vtk.py
+++ b/examples/cfd/external_aerodynamics/transformer_models/src/inference_on_vtk.py
@@ -304,10 +304,10 @@ def build_data_dict(
 
     # Add flow parameters
     data_dict["air_density"] = torch.tensor(
-        [air_density], device=device, dtype=torch.float32
+        air_density, device=device, dtype=torch.float32
     )
     data_dict["stream_velocity"] = torch.tensor(
-        [stream_velocity], device=device, dtype=torch.float32
+        stream_velocity, device=device, dtype=torch.float32
     )
 
     return data_dict
diff --git a/physicsnemo/experimental/models/geotransolver/gale.py b/physicsnemo/experimental/models/geotransolver/gale.py
index 6990ece8fa..b64f8e9df5 100644
--- a/physicsnemo/experimental/models/geotransolver/gale.py
+++ b/physicsnemo/experimental/models/geotransolver/gale.py
@@ -457,7 +457,7 @@ def forward(
         attn = self.Attn(tuple(normed_inputs), global_context)
 
         # Residual connection after attention
-        fx_out = [attn[i] + normed_inputs[i] for i in range(len(normed_inputs))]
+        fx_out = [attn[i] + fx[i] for i in range(len(fx))]
 
         # Feed-forward network with residual connection
         fx_out = [self.ln_mlp1(_fx) + _fx for _fx in fx_out]
diff --git a/test/models/geotransolver/data/geotransolver_basic_output.pth b/test/models/geotransolver/data/geotransolver_basic_output.pth
index 4c616db8a6..591bc6aa74 100644
Binary files a/test/models/geotransolver/data/geotransolver_basic_output.pth and b/test/models/geotransolver/data/geotransolver_basic_output.pth differ
diff --git a/test/models/geotransolver/data/geotransolver_te_output.pth b/test/models/geotransolver/data/geotransolver_te_output.pth
deleted file mode 100644
index 6b85b2324d..0000000000
Binary files a/test/models/geotransolver/data/geotransolver_te_output.pth and /dev/null differ
diff --git a/test/models/geotransolver/data/geotransolver_tuple_output.pth b/test/models/geotransolver/data/geotransolver_tuple_output.pth
index 63329a08eb..16aa08a036 100644
Binary files a/test/models/geotransolver/data/geotransolver_tuple_output.pth and b/test/models/geotransolver/data/geotransolver_tuple_output.pth differ