             // TODO: delegate
         } else if toController is PhotoViewController {
             
-            let photoCell = self.tableView.cellForRowAtIndexPath(NSIndexPath(forRow: 0, inSection: 2)) as? PhotoCell
+            let photoCell = self.tableView.cellForRowAtIndexPath(NSIndexPath(forRow: 0, inSection: 3)) as? PhotoCell
             if let photoCell = photoCell {
                 if photoCell.photoView.image == nil {
                     Helper.sharedInstance.showActionSheet(self, imagePicker: imagePicker)

        isCollaped = true
        
        addBarButton()

        if currentAccount == nil {
            currentAccount = Account.defaultAccount()
        }

        if selectedTransaction != nil {
            navigationItem.title = "Edit Transaction"
        } else {

        
        if isShowDatePicker {
            isShowDatePicker = false
            tableView.reloadSections(NSIndexSet(index: 2), withRowAnimation: UITableViewRowAnimation.Automatic)
        } else {
            isShowDatePicker = true
            tableView.reloadSections(NSIndexSet(index: 2), withRowAnimation: UITableViewRowAnimation.Automatic)
        }
    }
    
