diff --git a/idea/testData/script/definition/highlighting/asyncResolver/template/template.kt b/idea/testData/script/definition/highlighting/asyncResolver/template/template.kt
index 9546b14b545..1c9ec642c1e 100644
--- a/idea/testData/script/definition/highlighting/asyncResolver/template/template.kt
+++ b/idea/testData/script/definition/highlighting/asyncResolver/template/template.kt
@@ -4,6 +4,7 @@ import kotlin.script.dependencies.*
 import kotlin.script.experimental.dependencies.*
 import kotlin.script.templates.*
 import java.io.File
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : AsyncDependenciesResolver {
     suspend override fun resolveAsync(scriptContents: ScriptContents, environment: Environment): DependenciesResolver.ResolveResult {
@@ -13,6 +14,7 @@ class TestDependenciesResolver : AsyncDependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 class Template : Base()
 
diff --git a/idea/testData/script/definition/highlighting/customBaseClass/template/template.kt b/idea/testData/script/definition/highlighting/customBaseClass/template/template.kt
index 86dc9f8e978..17aa2b32064 100644
--- a/idea/testData/script/definition/highlighting/customBaseClass/template/template.kt
+++ b/idea/testData/script/definition/highlighting/customBaseClass/template/template.kt
@@ -5,6 +5,7 @@ import kotlin.script.templates.*
 import java.io.File
 import java.util.concurrent.CompletableFuture
 import java.util.concurrent.Future
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : ScriptDependenciesResolver {
     override fun resolve(
@@ -20,6 +21,7 @@ class TestDependenciesResolver : ScriptDependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 class Template: Base()
 
diff --git a/idea/testData/script/definition/highlighting/doNotSpeakAboutJava/template/template.kt b/idea/testData/script/definition/highlighting/doNotSpeakAboutJava/template/template.kt
index c8cd5917c74..57dad8467fa 100644
--- a/idea/testData/script/definition/highlighting/doNotSpeakAboutJava/template/template.kt
+++ b/idea/testData/script/definition/highlighting/doNotSpeakAboutJava/template/template.kt
@@ -4,6 +4,7 @@ import kotlin.script.dependencies.*
 import kotlin.script.experimental.dependencies.*
 import kotlin.script.templates.*
 import java.io.File
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : DependenciesResolver {
     override fun resolve(scriptContents: ScriptContents, environment: Environment): DependenciesResolver.ResolveResult {
@@ -41,6 +42,7 @@ class TestDependenciesResolver : DependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 class Template : Base()
 
diff --git a/idea/testData/script/definition/highlighting/doNotSpeakAboutJavaLegacy/template/template.kt b/idea/testData/script/definition/highlighting/doNotSpeakAboutJavaLegacy/template/template.kt
index 7c3fdace6d0..27eadc53835 100644
--- a/idea/testData/script/definition/highlighting/doNotSpeakAboutJavaLegacy/template/template.kt
+++ b/idea/testData/script/definition/highlighting/doNotSpeakAboutJavaLegacy/template/template.kt
@@ -5,6 +5,7 @@ import kotlin.script.templates.*
 import java.io.File
 import java.util.concurrent.CompletableFuture
 import java.util.concurrent.Future
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : ScriptDependenciesResolver {
     override fun resolve(
@@ -35,6 +36,7 @@ class TestDependenciesResolver : ScriptDependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 class Template : Base()
 
diff --git a/idea/testData/script/definition/highlighting/emptyAsyncResolver/template/template.kt b/idea/testData/script/definition/highlighting/emptyAsyncResolver/template/template.kt
index b813da41e7e..4452a202ad4 100644
--- a/idea/testData/script/definition/highlighting/emptyAsyncResolver/template/template.kt
+++ b/idea/testData/script/definition/highlighting/emptyAsyncResolver/template/template.kt
@@ -4,6 +4,7 @@ import java.io.File
 import kotlin.script.dependencies.*
 import kotlin.script.experimental.dependencies.*
 import kotlin.script.templates.ScriptTemplateDefinition
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : AsyncDependenciesResolver {
     override suspend fun resolveAsync(
@@ -14,6 +15,7 @@ class TestDependenciesResolver : AsyncDependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 open class Template: Base()
 
diff --git a/idea/testData/script/definition/highlighting/errorResolver/template/template.kt b/idea/testData/script/definition/highlighting/errorResolver/template/template.kt
index 667365afc62..dd14ba6b7e5 100644
--- a/idea/testData/script/definition/highlighting/errorResolver/template/template.kt
+++ b/idea/testData/script/definition/highlighting/errorResolver/template/template.kt
@@ -4,6 +4,7 @@ import java.io.File
 import kotlin.script.dependencies.*
 import kotlin.script.experimental.dependencies.*
 import kotlin.script.templates.ScriptTemplateDefinition
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : DependenciesResolver {
     override fun resolve(
@@ -20,5 +21,6 @@ class TestDependenciesResolver : DependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 open class Template
\ No newline at end of file
diff --git a/idea/testData/script/definition/highlighting/noResolver/template/template.kt b/idea/testData/script/definition/highlighting/noResolver/template/template.kt
index db1336fe7fd..df93336a55d 100644
--- a/idea/testData/script/definition/highlighting/noResolver/template/template.kt
+++ b/idea/testData/script/definition/highlighting/noResolver/template/template.kt
@@ -3,7 +3,9 @@ package custom.scriptDefinition
 import java.io.File
 import kotlin.script.dependencies.*
 import kotlin.script.templates.ScriptTemplateDefinition
+import kotlin.script.experimental.location.*
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(scriptFilePattern = "script.kts")
 class Template: Base()
 
diff --git a/idea/testData/script/definition/highlighting/throwingResolver/template/template.kt b/idea/testData/script/definition/highlighting/throwingResolver/template/template.kt
index 4b0191adeb5..e4ccca4005b 100644
--- a/idea/testData/script/definition/highlighting/throwingResolver/template/template.kt
+++ b/idea/testData/script/definition/highlighting/throwingResolver/template/template.kt
@@ -4,6 +4,7 @@ import java.io.File
 import kotlin.script.dependencies.*
 import kotlin.script.experimental.dependencies.*
 import kotlin.script.templates.ScriptTemplateDefinition
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : DependenciesResolver {
     override fun resolve(
@@ -14,5 +15,6 @@ class TestDependenciesResolver : DependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 open class Template
\ No newline at end of file