diff --git a/src/Setup_geometry.jl b/src/Setup_geometry.jl
index bb40a93d..2c88c160 100644
--- a/src/Setup_geometry.jl
+++ b/src/Setup_geometry.jl
@@ -299,7 +299,7 @@ function add_box!(Phase, Temp, Grid::AbstractGeneralGrid;       # required input
                 Phase[ind_flat] = compute_phase(Phase[ind_flat], Temp[ind_flat], Xrot[ind], Yrot[ind], Zrot[ind], phase)
             end
             if segments !== nothing
-                Temp[ind_flat] = compute_thermal_structure(Temp[ind_flat], X[ind], Y[ind], Z[ind], Phase[ind_flat], T, segments)
+                Temp[ind_flat] = compute_thermal_structure(Temp[ind_flat], Xrot[ind], Yrot[ind], Zrot[ind], Phase[ind_flat], T, segments)
             else
                 Temp[ind_flat] = compute_thermal_structure(Temp[ind_flat], Xrot[ind], Yrot[ind], Zrot[ind], Phase[ind_flat], T)
             end