diff --git a/diktat-rules/src/test/kotlin/org/cqfn/diktat/ruleset/chapter3/spaces/IndentationRuleWarnTest.kt b/diktat-rules/src/test/kotlin/org/cqfn/diktat/ruleset/chapter3/spaces/IndentationRuleWarnTest.kt
index 72d081d508..8b971f6ea9 100644
--- a/diktat-rules/src/test/kotlin/org/cqfn/diktat/ruleset/chapter3/spaces/IndentationRuleWarnTest.kt
+++ b/diktat-rules/src/test/kotlin/org/cqfn/diktat/ruleset/chapter3/spaces/IndentationRuleWarnTest.kt
@@ -396,19 +396,19 @@ class IndentationRuleWarnTest : LintTestBase(::IndentationRule) {
             """
                     |fun foo() {
                     |    consume(Example(
-                    |            t1, t2, t3
+                    |        t1, t2, t3
                     |    ))
                     |
                     |    bar(baz(
-                    |            1,
-                    |            2
+                    |        1,
+                    |        2
                     |    )
                     |    )
                     |
                     |    bar(baz(
-                    |            1,
-                    |            2),
-                    |            3
+                    |        1,
+                    |        2),
+                    |        3
                     |    )
                     |}
                     |
@@ -484,15 +484,15 @@ class IndentationRuleWarnTest : LintTestBase(::IndentationRule) {
             """
                     |fun foo() {
                     |    bar(
-                    |            param1 = baz(
-                    |                    1,
-                    |                    2
-                    |            ),
-                    |            param2 = { elem ->
-                    |                elem.qux()
-                    |            },
-                    |            param3 = x
-                    |                    .y()
+                    |        param1 = baz(
+                    |            1,
+                    |            2
+                    |        ),
+                    |        param2 = { elem ->
+                    |            elem.qux()
+                    |        },
+                    |        param3 = x
+                    |                .y()
                     |    )
                     |}
                     |
diff --git a/diktat-rules/src/test/kotlin/org/cqfn/diktat/util/LintTestBase.kt b/diktat-rules/src/test/kotlin/org/cqfn/diktat/util/LintTestBase.kt
index bd53570716..512e8c17cf 100644
--- a/diktat-rules/src/test/kotlin/org/cqfn/diktat/util/LintTestBase.kt
+++ b/diktat-rules/src/test/kotlin/org/cqfn/diktat/util/LintTestBase.kt
@@ -6,6 +6,7 @@ import com.pinterest.ktlint.core.KtLint
 import com.pinterest.ktlint.core.LintError
 import com.pinterest.ktlint.core.Rule
 import com.pinterest.ktlint.core.api.FeatureInAlphaState
+import org.intellij.lang.annotations.Language
 
 /**
  * Base class for testing rules without fixing code.
@@ -23,7 +24,7 @@ open class LintTestBase(private val ruleSupplier: (rulesConfigList: List<RulesCo
      * @param fileName optional override for file name
      */
     @OptIn(FeatureInAlphaState::class)
-    fun lintMethod(code: String,
+    fun lintMethod(@Language("kotlin") code: String,
                    vararg lintErrors: LintError,
                    rulesConfigList: List<RulesConfig>? = null,
                    fileName: String? = null