Fix masonry layout