diff --git a/integration-tests/src/__test__/examples.spec.ts b/integration-tests/src/__test__/examples.spec.ts
index bbcd2cf94..c3fb8f5f0 100644
--- a/integration-tests/src/__test__/examples.spec.ts
+++ b/integration-tests/src/__test__/examples.spec.ts
@@ -877,7 +877,7 @@ describe("Testing examples", () => {
     const [resEmpty, resFirst, resSecond] = await testKeysFuncCall();
     expect(resEmpty).toEqual([]);
     expect(resFirst).toEqual(["key one"]);
-    expect(resSecond).toEqual(["key one", "key two"]);
+    expect(resSecond).toEqual(["key one", "key two", "key one", "key one"]);
   });
 
   it("streamMap.aqua keysStream function call", async () => {
diff --git a/model/inline/src/main/scala/aqua/model/inline/raw/ApplyStreamMapRawInliner.scala b/model/inline/src/main/scala/aqua/model/inline/raw/ApplyStreamMapRawInliner.scala
index c0bfa116d..c089b7cb6 100644
--- a/model/inline/src/main/scala/aqua/model/inline/raw/ApplyStreamMapRawInliner.scala
+++ b/model/inline/src/main/scala/aqua/model/inline/raw/ApplyStreamMapRawInliner.scala
@@ -157,7 +157,7 @@ object ApplyStreamMapRawInliner {
       CanonicalizeModel(mapVar, CallModel.Export(canonMap.name, canonMap.`type`)).leaf,
       idxModel,
       FlattenModel(
-        canonMap.withProperty(IntoIndexModel(idx.name, CanonStreamType(mapType.element))),
+        canonMap.withProperty(IntoIndexModel(idx.name, ArrayType(mapType.element))),
         resultName
       ).leaf
     )