diff --git a/line-breaking.js b/line-breaking.js
index b07de52..53f8375 100644
--- a/line-breaking.js
+++ b/line-breaking.js
@@ -18,8 +18,8 @@ window.addEventListener("load", () => {
 
 window.addEventListener("message", async event => {
     if (!event?.data) return
-    if (event.data.type === "SELECT_ANNOTATION" || event.data.type === "CURRENT_LINE_INDEX") {
-        document.getElementById("insert-text").dataset.lineId = event.data.lineId
+    if (event.data.type === "UPDATE_CURRENT_LINE") {
+        document.getElementById("insert-text").dataset.lineId = event.data.currentLineId
     }
 })