             }
         }
         
-        self.displayMissingCells(self.scrollView.layoutMode)
+        self.displayMissingCells()
     }
     
     //MARK: - Editing mode

                            UIView.animateWithDuration(0.3, animations: { () -> Void in
                                self.scrollView.adjustContentSize()
                            }, completion: { (_) -> Void in
                                self.displayMissingCells(self.scrollView.layoutMode)
                            })
                        })
                    })

    }
    
    override func didRotateFromInterfaceOrientation(fromInterfaceOrientation: UIInterfaceOrientation) {
        self.displayMissingCells()
    }
    
    //MARK: - ScrollView delegate
    
    override func handleSnapBack() {
        self.displayMissingCells()
    }
    
    func scrollViewDidEndDecelerating(scrollView: UIScrollView) {
