diff --git a/masonry_views.module b/masonry_views.module
index 4ebe8a0..ce6bc1c 100644
--- a/masonry_views.module
+++ b/masonry_views.module
@@ -21,7 +21,8 @@ function masonry_views_preprocess_views_view_masonry(&$variables) {
   $options = $style->options;
 
   // Build the view container
-  $container = '.view-' . Html::cleanCssIdentifier($view->storage->id()) . '.view-display-id-' . $view->current_display . ' > .view-content';
+  //$container = '.view-' . Html::cleanCssIdentifier($view->storage->id()) . '.view-display-id-' . $view->current_display . ' > .view-content';
+  $container = '.masonry-layout';
 
   // If grouping enabled, style it.
   if (!empty($options['grouping'])) {
diff --git a/templates/views-view-masonry.html.twig b/templates/views-view-masonry.html.twig
index 4771e46..4fd6f79 100644
--- a/templates/views-view-masonry.html.twig
+++ b/templates/views-view-masonry.html.twig
@@ -20,22 +20,24 @@
  */
 #}
 
-{% if title and grouping %}
-  <h3>{{ title }}</h3>
-  <div class="{{grouping_class}}">
-{% endif %}
+<div class="masonry-layout">
+  {% if title and grouping %}
+    <h3>{{ title }}</h3>
+    <div class="{{grouping_class}}">
+  {% endif %}
 
-{% for row in rows %}
-  {%
-  set row_classes = [
-  default_row_class ? 'views-row',
-  ]
-  %}
-  <div{{ row.attributes.addClass(row_classes) }}>
-    {{ row.content }}
-  </div>
-{% endfor %}
+  {% for row in rows %}
+    {%
+    set row_classes = [
+    default_row_class ? 'views-row',
+    ]
+    %}
+    <div{{ row.attributes.addClass(row_classes) }}>
+      {{ row.content }}
+    </div>
+  {% endfor %}
 
-{% if grouping %}
-  </div>
-{% endif %}
+  {% if grouping %}
+    </div>
+  {% endif %}
+</div>
\ No newline at end of file