diff --git a/floris/simulation/farm.py b/floris/simulation/farm.py
index 6e1c584d4..ce289ace2 100644
--- a/floris/simulation/farm.py
+++ b/floris/simulation/farm.py
@@ -503,7 +503,11 @@ def finalize(self, unsorted_indices):
     @property
     def coordinates(self):
         return np.array([
-            np.array([x, y, z]) for x, y, z in zip(self.layout_x, self.layout_y, self.hub_heights)
+            np.array([x, y, z]) for x, y, z in zip(
+                self.layout_x,
+                self.layout_y,
+                self.hub_heights if len(self.hub_heights.shape) == 1 else self.hub_heights[0,0]
+            )
         ])
 
     @property