diff --git a/js/masonry-init.js b/js/masonry-init.js
index 0cc1b42..2f1ab5b 100755
--- a/js/masonry-init.js
+++ b/js/masonry-init.js
@@ -1,7 +1,10 @@
-
-var elem = document.querySelector('.view-content');
+//==========================================
+var masonry_grid_container = '.container_'
+var masonry_grid_item = '.grid-item_'
+//==========================================
+var elem = document.querySelector(masonry_grid_container);
 var msnry = new Masonry( elem, {
   // options
-  itemSelector: '.views-row',
+  itemSelector: masonry_grid_item,
   columnWidth: 400
 });