diff --git a/idea/testData/dslHighlighter/functionCalls.kt b/idea/testData/dslHighlighter/functionCalls.kt
index b8e300fb8df..2b543c54673 100644
--- a/idea/testData/dslHighlighter/functionCalls.kt
+++ b/idea/testData/dslHighlighter/functionCalls.kt
@@ -21,6 +21,12 @@ fun g() {
 
 }
 
+@A
+class AC {
+    @A
+    fun a() = ""
+}
+
 fun test() {
     f() // 4
 
@@ -31,4 +37,8 @@ fun test() {
     g() // 1
 
     ff() // 4
+
+    with (A()) {
+        a() // 4
+    }
 }
\ No newline at end of file
diff --git a/idea/testData/dslHighlighter/propertyAccess.kt b/idea/testData/dslHighlighter/propertyAccess.kt
index 9b53fc86acc..50213f5b5f1 100644
--- a/idea/testData/dslHighlighter/propertyAccess.kt
+++ b/idea/testData/dslHighlighter/propertyAccess.kt
@@ -24,10 +24,16 @@ var BC.p7
     set(i) {}
 
 @A
-class AC
+class AC {
+    @A
+    val a = ""
+}
 
 @B
-class BC
+class BC {
+    @B
+    val b = ""
+}
 
 fun test() {
     p2 // 4
@@ -35,10 +41,12 @@ fun test() {
 
     with(AC()) {
         p1 // 4
+        a // 4
     }
     with(BC()) {
         p3 // 1
         p7 // 1
         p7 = 3 // 1
+        b // 1
     }
 }
\ No newline at end of file