diff --git a/packages/html/stories/Folding.stories.js b/packages/html/stories/Folding.stories.js
index de875d15b..f730ab489 100644
--- a/packages/html/stories/Folding.stories.js
+++ b/packages/html/stories/Folding.stories.js
@@ -83,7 +83,7 @@ const Template = ({ label, ...args }) => {
   // Adds cells to the model in a single step
   graph.batchUpdate(() => {
     const col1 = graph.insertVertex(parent, null, '', 0, 0, 120, 0, {
-      baseStyleName: 'column',
+      baseStyleNames: ['column'],
     });
 
     const v1 = graph.insertVertex(col1, null, '1', 0, 0, 100, 30);
@@ -98,7 +98,7 @@ const Template = ({ label, ...args }) => {
     const v12 = graph.insertVertex(v1, null, '1.2', 0, 0, 80, 30);
 
     const col2 = graph.insertVertex(parent, null, '', 0, 0, 120, 0, {
-      baseStyleName: 'column',
+      baseStyleNames: ['column'],
     });
 
     const v2 = graph.insertVertex(col2, null, '2', 0, 0, 100, 30);