diff --git a/webui/src/main/kotlin/com/simiacryptus/diff/AddApplyFileDiffLinks.kt b/webui/src/main/kotlin/com/simiacryptus/diff/AddApplyFileDiffLinks.kt
index c6e15d5a..3fd8b8c7 100644
--- a/webui/src/main/kotlin/com/simiacryptus/diff/AddApplyFileDiffLinks.kt
+++ b/webui/src/main/kotlin/com/simiacryptus/diff/AddApplyFileDiffLinks.kt
@@ -53,11 +53,11 @@ fun SocketManagerBase.addApplyFileDiffLinks(
   val findAll = codeblockPattern.findAll(response).toList()
   val codeblocks = findAll.filter { block ->
     try {
-      val header = headers.lastOrNull { it.first.last <= block.range.first } ?: defaultFile?.let { IntRange(0,0) to it }
+      val header = headers.lastOrNull { it.first.last <= block.range.first }?.let { it.second } ?: defaultFile
       if (header == null) {
         return@filter false
       }
-      val filename = resolve(root, header.second)
+      val filename = resolve(root, header)
       !root.resolve(filename).toFile().exists()
     } catch (e: Throwable) {
       log.info("Error processing code block", e)
@@ -66,11 +66,11 @@ fun SocketManagerBase.addApplyFileDiffLinks(
   }.map { it.range to it }.toList()
   val patchBlocks = findAll.filter { block ->
     try {
-      val header = headers.lastOrNull { it.first.last <= block.range.first } ?: defaultFile?.let { IntRange(0,0) to it }
+      val header = headers.lastOrNull { it.first.last <= block.range.first }?.let { it.second } ?: defaultFile
       if (header == null) {
         return@filter false
       }
-      val filename = resolve(root, header.second)
+      val filename = resolve(root, header)
       root.resolve(filename).toFile().exists()
     } catch (e: Throwable) {
       log.info("Error processing code block", e)
@@ -127,7 +127,7 @@ fun SocketManagerBase.addApplyFileDiffLinks(
   // Process diff blocks and add patch links
   val withPatchLinks: String = patchBlocks.foldIndexed(response) { index, markdown, diffBlock ->
     val value = diffBlock.second.groupValues[2].trim()
-    var header = headers.lastOrNull { it.first.last < diffBlock.first.first }?.second ?: "Unknown"
+    var header = headers.lastOrNull { it.first.last < diffBlock.first.first }?.second ?: defaultFile ?: "Unknown"
     header = corrections?.get("patch_$index") ?: header
     val filename = resolve(root, header)
     val newValue = renderDiffBlock(root, filename, value, handle, ui, api, shouldAutoApply)
@@ -137,7 +137,7 @@ fun SocketManagerBase.addApplyFileDiffLinks(
   val withSaveLinks = codeblocks.foldIndexed(withPatchLinks) { index, markdown, codeBlock ->
     val lang = codeBlock.second.groupValues[1]
     val value = codeBlock.second.groupValues[2].trim()
-    var header = headers.lastOrNull { it.first.last < codeBlock.first.first }?.second
+    var header = headers.lastOrNull { it.first.last < codeBlock.first.first }?.second ?: defaultFile ?: "Unknown"
     header = corrections?.get("code_$index") ?: header
     val newMarkdown = renderNewFile(header, root, ui, shouldAutoApply, value, handle, lang)
     markdown.replace(codeBlock.second.value, newMarkdown)
diff --git a/webui/src/main/kotlin/com/simiacryptus/skyenet/TabbedDisplay.kt b/webui/src/main/kotlin/com/simiacryptus/skyenet/TabbedDisplay.kt
index 719d1733..2a67eaf9 100644
--- a/webui/src/main/kotlin/com/simiacryptus/skyenet/TabbedDisplay.kt
+++ b/webui/src/main/kotlin/com/simiacryptus/skyenet/TabbedDisplay.kt
@@ -15,7 +15,7 @@ open class TabbedDisplay(
 
   val size: Int get() = tabs.size
   val tabId = UUID.randomUUID()
-  open fun render() = if (tabs.isEmpty()) "<div/>" else {
+  private fun render() = if (tabs.isEmpty()) "<div/>" else {
     """
   <div class="tabs-container" id="$tabId">
   ${renderTabButtons()}
@@ -32,7 +32,7 @@ open class TabbedDisplay(
     task.add(render())!!
   }
 
-  open fun renderTabButtons() = """
+  protected open fun renderTabButtons() = """
 <div class="tabs">${
     tabs.toTypedArray().withIndex().joinToString("\n") { (idx, pair) ->
       if (idx == selectedTab) {
@@ -44,7 +44,7 @@ open class TabbedDisplay(
   }</div>
 """
 
-  open fun renderContentTab(t: Pair<String, StringBuilder>, idx: Int) = """
+  protected open fun renderContentTab(t: Pair<String, StringBuilder>, idx: Int) = """
 <div class="tab-content ${
     when {
       idx == selectedTab -> "active"