   /// - Returns: The index path for the note.
   private func indexPathForNote(atIndex index: Int) -> IndexPath {
     var itemIndex = index
-    if shouldAllowNoteAdditions {
+    if shouldShowAddNoteCell {
       itemIndex += 1
     }
     return IndexPath(item: itemIndex, section: Section.chartAndNotes.rawValue)

    case .header: return 1
    case .chartAndNotes:
      // The notes, plus one for the add note to timeline button, if additions are allowed.
      return displayTrial.notes.count + (shouldAllowNoteAdditions ? 1 : 0)
    }
  }


  /// - Parameter indexPath: An index path.
  /// - Returns: True if the index path is add note, otherwise false.
  func isAddNoteIndexPath(_ indexPath: IndexPath) -> Bool {
    return shouldShowAddNoteCell &&
      indexPath.section == Section.chartAndNotes.rawValue &&
      indexPath.item == 0
  }
