diff --git a/binaries/data/mods/public/simulation/ai/common-api/shared.js b/binaries/data/mods/public/simulation/ai/common-api/shared.js
index 7027c3d345..9bd0d8f329 100644
--- a/binaries/data/mods/public/simulation/ai/common-api/shared.js
+++ b/binaries/data/mods/public/simulation/ai/common-api/shared.js
@@ -148,7 +148,7 @@ m.SharedScript.prototype.init = function(state, deserialization)
 	var landPassMap = new Uint8Array(this.passabilityMap.data.length);
 	var waterPassMap = new Uint8Array(this.passabilityMap.data.length);
 	var obstructionMaskLand = this.passabilityClasses["default-terrain-only"];
-	var obstructionMaskWater = this.passabilityClasses["ship-small"];
+	var obstructionMaskWater = this.passabilityClasses["ship-terrain-only"];
 	for (var i = 0; i < this.passabilityMap.data.length; ++i)
 	{
 		landPassMap[i] = (this.passabilityMap.data[i] & obstructionMaskLand) ? 0 : 255;
diff --git a/binaries/data/mods/public/simulation/ai/common-api/terrain-analysis.js b/binaries/data/mods/public/simulation/ai/common-api/terrain-analysis.js
index 98ae132a11..7982a30f5f 100644
--- a/binaries/data/mods/public/simulation/ai/common-api/terrain-analysis.js
+++ b/binaries/data/mods/public/simulation/ai/common-api/terrain-analysis.js
@@ -29,7 +29,7 @@ m.TerrainAnalysis.prototype.init = function(sharedScript,rawState)
 	this.cellSize = passabilityMap.cellSize;
 
 	var obstructionMaskLand = rawState.passabilityClasses["default-terrain-only"];
-	var obstructionMaskWater = rawState.passabilityClasses["ship-small"];
+	var obstructionMaskWater = rawState.passabilityClasses["ship-terrain-only"];
 
 	var obstructionTiles = new Uint8Array(passabilityMap.data.length);