diff --git a/prefabs/units/army/army.gd b/prefabs/units/army/army.gd
index a499943..ac1c2d2 100644
--- a/prefabs/units/army/army.gd
+++ b/prefabs/units/army/army.gd
@@ -40,6 +40,9 @@ func _process(delta):
 						resolve_conflict(grid.grid[mouse_pos].units)
 					else:
 						grid.set_country(mouse_pos, country)
+					
+					if grid.grid[pos].needs_boat:
+						grid.set_country(pos, grid.countries[0])
 
 func resolve_conflict(armies):
 	var survivor = armies[randi() % len(armies)]