diff --git a/IV/tests-integration.sql b/IV/tests-integration.sql
index 3d93e8b..cc431ee 100644
--- a/IV/tests-integration.sql
+++ b/IV/tests-integration.sql
@@ -2,4 +2,4 @@
 
 drop table if exists agg_rivers_wm;
 create table agg_rivers_wm (name text, way geometry);
-insert into agg_rivers_wm (name, way) select name, ST_SimplifyWM_DEBUG(way) from agg_rivers where name='Visinčia';
+insert into agg_rivers_wm (name, way) select name, ST_SimplifyWM_DEBUG(way) from agg_rivers;
diff --git a/IV/wm.sql b/IV/wm.sql
index 1541a13..83619e3 100644
--- a/IV/wm.sql
+++ b/IV/wm.sql
@@ -298,7 +298,7 @@ begin
   end loop;
 
   if l_type = 'ST_LineString' then
-    return bends[1];
+    return st_linemerge(st_union(bends));
   elseif l_type = 'ST_MultiLineString' then
     return st_union(bends);
   end if;
@@ -351,7 +351,7 @@ begin
   end loop;
 
   if l_type = 'ST_LineString' then
-    return bends[1];
+    return st_linemerge(st_union(bends));
   elseif l_type = 'ST_MultiLineString' then
     return st_union(bends);
   end if;