         }
         
         let tag = Tag(text: tags[indexPath.item], index: indexPath.item)
-        prototypeCell.configure(with: tag)
-        let size = prototypeCell.wmf_preferredFrame(at: .zero, fitting: prototypeCell.width, alignedBy: semanticContentAttribute, apply: false).size
+        placeholderCell.configure(with: tag)
+        let size = placeholderCell.wmf_preferredFrame(at: .zero, fitting: placeholderCell.width, alignedBy: semanticContentAttribute, apply: false).size
         tagsCollectionViewWidth += size.width
+        layout?.itemSize = size
         return size
     }
 }

        let tagsCount: CGFloat = CGFloat(tags.count)
        if (apply && tagsCount != 0), let layout = layout {
            let width = tagsCollectionViewWidth + (layout.minimumLineSpacing * (tagsCount - 1))
            
            collectionView.frame = CGRect(x: layoutMargins.left, y: origin.y, width: width, height: layout.itemSize.height)
            collectionView.backgroundColor = UIColor.cyan
        }

        return layout
    }()
    
    fileprivate lazy var placeholderCell: TagCollectionViewCell = {
        return TagCollectionViewCell()
    }()
    
