diff --git a/third_party/src/main/java/com/jetbrains/lang/dart/ide/template/DartTemplateContextType.java b/third_party/src/main/java/com/jetbrains/lang/dart/ide/template/DartTemplateContextType.java
index 65cbd895e..c3a507bee 100644
--- a/third_party/src/main/java/com/jetbrains/lang/dart/ide/template/DartTemplateContextType.java
+++ b/third_party/src/main/java/com/jetbrains/lang/dart/ide/template/DartTemplateContextType.java
@@ -1,6 +1,7 @@
 // Copyright 2000-2024 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
 package com.jetbrains.lang.dart.ide.template;
 
+import com.intellij.codeInsight.template.TemplateActionContext;
 import com.intellij.codeInsight.template.TemplateContextType;
 import com.intellij.openapi.fileTypes.SyntaxHighlighter;
 import com.intellij.openapi.util.NlsContexts;
@@ -32,9 +33,10 @@ protected DartTemplateContextType(@NotNull @NonNls String id,
   }
 
   @Override
-  public boolean isInContext(@NotNull PsiFile file, int offset) {
+  public boolean isInContext(@NotNull TemplateActionContext templateActionContext) {
+    PsiFile file = templateActionContext.getFile();
     if (file.getLanguage() instanceof DartLanguage) {
-      PsiElement element = file.findElementAt(offset);
+      PsiElement element = file.findElementAt(templateActionContext.getStartOffset());
       return element != null && isInContext(element);
     }
     return false;