     }
 
     init() {
-        flowLayout = UICollectionViewFlowLayout()
+        flowLayout = FlowLayout()
         super.init(collectionViewLayout: flowLayout)
         title = CommonStrings.searchTitle
     }

    override func viewDidLoad() {
        super.viewDidLoad()
        collectionView.register(InsertMediaSearchResultCollectionViewCell.self, forCellWithReuseIdentifier: InsertMediaSearchResultCollectionViewCell.identifier)
        flowLayout.itemSize = CGSize(width: itemDimension, height: itemDimension)
        flowLayout.minimumInteritemSpacing = 8
        flowLayout.minimumLineSpacing = 8
        // TODO: Make this VC a ViewController so that it can work with the nav bar
        collectionView.contentInset = UIEdgeInsets(top: 120, left: 12, bottom: 0, right: 24)
    }

    func reload() {

        guard viewIfLoaded != nil else {
            return
        }
        view.backgroundColor = theme.colors.paperBackground
        collectionView.backgroundColor = theme.colors.paperBackground
    }
}
