diff --git a/includes/MasonryWidget.php b/includes/MasonryWidget.php
index 0817931..545ba4a 100644
--- a/includes/MasonryWidget.php
+++ b/includes/MasonryWidget.php
@@ -61,14 +61,22 @@ CSS;
         $parser->getOutput()->addModules(['ext.isekai.masonry']);
 
         $params['cols'] = intval($params['cols'] ?? 2);
-        $params['xs-cols'] = intval($params['xs-cols'] ?? 2);
-        $params['sm-cols'] = intval($params['sm-cols'] ?? 2);
+        $params['xs-cols'] = intval($params['xs-cols'] ?? 1);
+        $params['sm-cols'] = intval($params['sm-cols'] ?? 1);
         $params['md-cols'] = intval($params['md-cols'] ?? 2);
         $params['lg-cols'] = intval($params['lg-cols'] ?? 2);
         $params['xl-cols'] = intval($params['xl-cols'] ?? 3);
         $params['gutter'] = intval($params['gutter'] ?? 10);
 
-        $className = ['isekai-masonry', "max-col-{$params['cols']}"];
+        $className = [
+            'isekai-masonry',
+            "max-col-{$params['cols']}",
+            "max-col-xs-{$params['xs-cols']}",
+            "max-col-sm-{$params['sm-cols']}",
+            "max-col-md-{$params['md-cols']}",
+            "max-col-lg-{$params['lg-cols']}",
+            "max-col-xl-{$params['xl-cols']}",
+        ];
 
         self::$paramsStack[] = $params;
         $content = $parser->recursiveTagParseFully($text, $frame);
@@ -79,11 +87,6 @@ CSS;
                 Html::openElement('div', [
                     'class' => implode(' ', $className),
                     'data-cols' => $params['cols'],
-                    'data-xs-cols' => $params['xs-cols'],
-                    'data-sm-cols' => $params['sm-cols'],
-                    'data-md-cols' => $params['md-cols'],
-                    'data-lg-cols' => $params['lg-cols'],
-                    'data-xl-cols' => $params['xl-cols'],
                     'data-gutter' => $params['gutter'],
                 ]) .
                 Html::element('div', [