From ad6f725d37e51a595990ccf0089aed6f27edfe38 Mon Sep 17 00:00:00 2001
From: Michael Steenbeek <m.o.steenbeek@gmail.com>
Date: Thu, 27 Jul 2017 17:50:01 +0200
Subject: [PATCH] Fix refactor error

---
 src/openrct2/world/map.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/openrct2/world/map.c b/src/openrct2/world/map.c
index c814fb732c..fd619eb505 100644
--- a/src/openrct2/world/map.c
+++ b/src/openrct2/world/map.c
@@ -1366,7 +1366,7 @@ static money32 map_change_surface_style(sint32 x0, sint32 y0, sint32 x1, sint32
 
             if (flags & 1)
             {
-                if (map_get_water_height(mapElement) == 0)
+                if (!(mapElement->properties.surface.terrain & MAP_ELEMENT_SURFACE_TERRAIN_MASK))
                 {
                     if (!(mapElement->type & MAP_ELEMENT_DIRECTION_MASK))
                     {