diff --git a/src/main.rs b/src/main.rs
index 663b21c..34c374f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -149,11 +149,10 @@ async fn main() -> Res<()> {
                             }
                         }
 
-                        if node
-                            .app_id
-                            .as_ref()
-                            .is_some_and(|app_id| app_id.eq_ignore_ascii_case("code-url-handler"))
-                        {
+                        if node.app_id.as_ref().is_some_and(|app_id| {
+                            app_id.eq_ignore_ascii_case("code-url-handler")
+                                || app_id.eq_ignore_ascii_case("code")
+                        }) {
                             recent_code = Some((std::time::Instant::now(), node.id));
                         } else if recent_code.is_some_and(|(t, _)| {
                             std::time::Instant::now().duration_since(t)