diff --git a/floris/core/farm.py b/floris/core/farm.py
index 6ab28d2a0..58b29637b 100644
--- a/floris/core/farm.py
+++ b/floris/core/farm.py
@@ -465,7 +465,7 @@ def coordinates(self):
             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]
+                self.hub_heights if len(self.hub_heights.shape) == 1 else self.hub_heights[0]
             )
         ])