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

  /// - Parameter indexPath: An index path.
  /// - Returns: True if the index path is a trial note in a valid range, otherwise false.
  func isTrialNoteIndexPath(_ indexPath: IndexPath) -> Bool {
    if shouldAllowNoteAdditions {
      return indexPath.section == Section.chartAndNotes.rawValue && indexPath.item > 0 &&
          indexPath.item <= displayTrial.notes.count
    } else {

    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 + (shouldShowAddNoteCell ? 1 : 0)
    }
  }

