diff --git a/src/main/kotlin/com/github/simiacryptus/aicoder/ui/TemperatureControlWidgetFactory.kt b/src/main/kotlin/com/github/simiacryptus/aicoder/ui/TemperatureControlWidgetFactory.kt
index 0553627e..6f183572 100644
--- a/src/main/kotlin/com/github/simiacryptus/aicoder/ui/TemperatureControlWidgetFactory.kt
+++ b/src/main/kotlin/com/github/simiacryptus/aicoder/ui/TemperatureControlWidgetFactory.kt
@@ -65,7 +65,7 @@ class TemperatureControlWidgetFactory : StatusBarWidgetFactory {
             return Consumer { event: MouseEvent ->
                 val widgetComp = event.component
                 if (widgetComp != null) {
-                    val modePanel = Panel()
+                    val modePanel = JPanel()
                     modePanel.layout = VerticalLayout(0)
                     modePanel.add(JBLabel("AI Coding Assistant"))
 
diff --git a/src/test/kotlin/com/github/simiacryptus/aicoder/util/ApxPatchUtilTest.kt b/src/test/kotlin/com/github/simiacryptus/aicoder/util/ApxPatchUtilTest.kt
index dbced010..85842250 100644
--- a/src/test/kotlin/com/github/simiacryptus/aicoder/util/ApxPatchUtilTest.kt
+++ b/src/test/kotlin/com/github/simiacryptus/aicoder/util/ApxPatchUtilTest.kt
@@ -1,6 +1,6 @@
 package com.github.simiacryptus.aicoder.util
 
-import com.github.simiacryptus.diff.ApxPatchUtil
+import com.github.simiacryptus.diff.PatchUtil
 import org.junit.Assert.assertTrue
 import org.junit.Test
 
@@ -70,7 +70,7 @@ class ApxPatchUtilTest {
 
   @Test
   fun testPatch() {
-    val result = ApxPatchUtil.patch(source, patch)
+    val result = PatchUtil.patch(source, patch)
     println(result)
     assertTrue(result.contains("for (value in values"))
   }