diff --git a/plugins/lombok/testData/compile/accessorsStripPrefix.kt b/plugins/lombok/testData/compile/accessorsStripPrefix.kt
index ecc57b7fc5a..a342f92ad91 100644
--- a/plugins/lombok/testData/compile/accessorsStripPrefix.kt
+++ b/plugins/lombok/testData/compile/accessorsStripPrefix.kt
@@ -3,8 +3,9 @@
 import lombok.*;
 import lombok.experimental.*;
 
-@Getter @Setter
-@Accessors(prefix = {"f", "field"})
+@Getter
+@Setter
+@Accessors(prefix = { "f", "field" })
 public class AccessorsTest {
 //    @Accessors
     private int age = 10;
@@ -17,8 +18,9 @@ public class AccessorsTest {
     @Accessors
     private Boolean isNonPrimitiveHuman;
 
-    static void test() {
-        val obj = new AccessorsTest();
+    static void test()
+    {
+        val obj = new AccessorsTest ();
 
 //        obj.getAge();
 //        obj.setAge(123);
@@ -44,28 +46,26 @@ public class AccessorsTest {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = AccessorsTest()
+fun box(): String {
+    val obj = AccessorsTest()
 
 //        obj.getAge()
 //        obj.setAge(123)
 
-        obj.getTarget()
-        obj.setTarget(34)
+    obj.getTarget()
+    obj.setTarget(34)
 
-        obj.getValue()
-        obj.setValue("sdf")
+    obj.getValue()
+    obj.setValue("sdf")
 
-        obj.isHuman()
-        obj.setHuman(true)
+    obj.isHuman()
+    obj.setHuman(true)
 
-        obj.isPrefixedBoolean()
-        obj.setPrefixedBoolean(false)
+    obj.isPrefixedBoolean()
+    obj.setPrefixedBoolean(false)
 
-        obj.getIsNonPrimitiveHuman()
-        obj.setIsNonPrimitiveHuman(false)
-
-    }
+    obj.getIsNonPrimitiveHuman()
+    obj.setIsNonPrimitiveHuman(false)
 
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/accessorsStripPrefixCombined.kt b/plugins/lombok/testData/compile/accessorsStripPrefixCombined.kt
index 9dc69812c90..cee060b8d08 100644
--- a/plugins/lombok/testData/compile/accessorsStripPrefixCombined.kt
+++ b/plugins/lombok/testData/compile/accessorsStripPrefixCombined.kt
@@ -15,15 +15,13 @@ public class PrefixJava {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        //not generated because doesn't have prefix from class level @Accessors
-        //assertEquals(PrefixJava().propA, "A")
-        //not generated because doesn't have prefix from config
-        //assertEquals(PrefixJava().propC, "C")
-        assertEquals(PrefixJava().propD, "D")
-    }
-
+fun box(): String {
+    //not generated because doesn't have prefix from class level @Accessors
+    //assertEquals(PrefixJava().propA, "A")
+    //not generated because doesn't have prefix from config
+    //assertEquals(PrefixJava().propC, "C")
+    assertEquals(PrefixJava().propD, "D")
+    return "OK"
 }
 
 //FILE: lombok.config
diff --git a/plugins/lombok/testData/compile/accessorsStripPrefixConfig.kt b/plugins/lombok/testData/compile/accessorsStripPrefixConfig.kt
index d6e3e267864..ee08fae9ab0 100644
--- a/plugins/lombok/testData/compile/accessorsStripPrefixConfig.kt
+++ b/plugins/lombok/testData/compile/accessorsStripPrefixConfig.kt
@@ -43,30 +43,27 @@ public class AccessorsTest {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = AccessorsTest()
+fun box(): String {
+    val obj = AccessorsTest()
 
-//        obj.getAge()
-//        obj.setAge(123)
+//    obj.getAge()
+//    obj.setAge(123)
 
-        obj.getTarget()
-        obj.setTarget(34)
+    obj.getTarget()
+    obj.setTarget(34)
 
-        obj.getValue()
-        obj.setValue("sdf")
+    obj.getValue()
+    obj.setValue("sdf")
 
-        obj.isHuman()
-        obj.setHuman(true)
+    obj.isHuman()
+    obj.setHuman(true)
 
-        obj.isPrefixedBoolean()
-        obj.setPrefixedBoolean(false)
-
-        obj.getIsNonPrimitiveHuman()
-        obj.setIsNonPrimitiveHuman(false)
-
-    }
+    obj.isPrefixedBoolean()
+    obj.setPrefixedBoolean(false)
 
+    obj.getIsNonPrimitiveHuman()
+    obj.setIsNonPrimitiveHuman(false)
+    return "OK"
 }
 
 //FILE: lombok.config
diff --git a/plugins/lombok/testData/compile/allArgsConstructor.kt b/plugins/lombok/testData/compile/allArgsConstructor.kt
index 7ed3d089441..e38af486f3c 100644
--- a/plugins/lombok/testData/compile/allArgsConstructor.kt
+++ b/plugins/lombok/testData/compile/allArgsConstructor.kt
@@ -19,8 +19,7 @@ public class ConstructorExample {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val generated = ConstructorExample(12, "sdf", true)
-    }
+fun box(): String {
+    val generated = ConstructorExample(12, "sdf", true)
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/allArgsConstructorStatic.kt b/plugins/lombok/testData/compile/allArgsConstructorStatic.kt
index 2febbd5295a..5af84360f94 100644
--- a/plugins/lombok/testData/compile/allArgsConstructorStatic.kt
+++ b/plugins/lombok/testData/compile/allArgsConstructorStatic.kt
@@ -24,9 +24,8 @@ public class ConstructorExample {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val existing: ConstructorExample = ConstructorExample("existing")
-        val generated: ConstructorExample = ConstructorExample.of(45, "234", false)
-    }
+fun box(): String {
+    val existing: ConstructorExample = ConstructorExample("existing")
+    val generated: ConstructorExample = ConstructorExample.of(45, "234", false)
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/clashAccessors.kt b/plugins/lombok/testData/compile/clashAccessors.kt
index 0b9169279b1..c0d1ecdde1c 100644
--- a/plugins/lombok/testData/compile/clashAccessors.kt
+++ b/plugins/lombok/testData/compile/clashAccessors.kt
@@ -79,32 +79,31 @@ class KotlinChildClass : ClashTest() {
 
 }
 
-class Test {
-    fun run() {
-        val obj = ClashTest()
+fun box(): String {
+    val obj = ClashTest()
 
-        obj.getAge()
-        //thats shouldn't work because lombok doesn't generate clashing method
+    obj.getAge()
+    //thats shouldn't work because lombok doesn't generate clashing method
 //        obj.setAge(41)
 //        obj.age = 12
-        val age = obj.age
+    val age = obj.age
 
 
-        obj.getName()
-        obj.setName("Al")
-        val name = obj.name
-        obj.name = "sdf"
+    obj.getName()
+    obj.setName("Al")
+    val name = obj.name
+    obj.name = "sdf"
 
-        obj.isHuman()
-        obj.setHuman(true)
-        obj.isHuman("sdf")
-        val isHuman = obj.isHuman
-        obj.isHuman = false
+    obj.isHuman()
+    obj.setHuman(true)
+    obj.isHuman("sdf")
+    val isHuman = obj.isHuman
+    obj.isHuman = false
 
-        val childObj = KotlinChildClass()
-        childObj.getToOverride()
-        childObj.setToOverride(34)
-        childObj.toOverride
-        childObj.toOverride = 412
-    }
+    val childObj = KotlinChildClass()
+    childObj.getToOverride()
+    childObj.setToOverride(34)
+    childObj.toOverride
+    childObj.toOverride = 412
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/configAccessors.kt b/plugins/lombok/testData/compile/configAccessors.kt
index d648d1c4453..731ea34bf00 100644
--- a/plugins/lombok/testData/compile/configAccessors.kt
+++ b/plugins/lombok/testData/compile/configAccessors.kt
@@ -19,13 +19,12 @@ public class GetterTest {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = GetterTest()
-        val ref: GetterTest = obj.name("some").age(34)
-        obj.name()
-        obj.age()
-    }
+fun box(): String {
+    val obj = GetterTest()
+    val ref: GetterTest = obj.name("some").age(34)
+    obj.name()
+    obj.age()
+   return "OK"
 }
 
 //FILE: lombok.config
diff --git a/plugins/lombok/testData/compile/configAccessorsOverride.kt b/plugins/lombok/testData/compile/configAccessorsOverride.kt
index 19667c15edd..eb02422f242 100644
--- a/plugins/lombok/testData/compile/configAccessorsOverride.kt
+++ b/plugins/lombok/testData/compile/configAccessorsOverride.kt
@@ -22,13 +22,12 @@ public class GetterTest {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = GetterTest()
-        val ref: GetterTest = obj.fluent(true)
-        obj.name()
-        obj.age()
-    }
+fun box(): String {
+    val obj = GetterTest()
+    val ref: GetterTest = obj.fluent(true)
+    obj.name()
+    obj.age()
+    return "OK"
 }
 
 //FILE: lombok.config
diff --git a/plugins/lombok/testData/compile/configCaseInsensitive.kt b/plugins/lombok/testData/compile/configCaseInsensitive.kt
index a713032ac91..168c574a9a3 100644
--- a/plugins/lombok/testData/compile/configCaseInsensitive.kt
+++ b/plugins/lombok/testData/compile/configCaseInsensitive.kt
@@ -15,13 +15,12 @@ public class GetterTest {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = GetterTest()
+fun box(): String {
+    val obj = GetterTest()
 
-        obj.primitiveBoolean
-        obj.getPrimitiveBoolean()
-    }
+    obj.primitiveBoolean
+    obj.getPrimitiveBoolean()
+    return "OK"
 }
 
 //FILE: lombok.config
diff --git a/plugins/lombok/testData/compile/configSimple.kt b/plugins/lombok/testData/compile/configSimple.kt
index ad516763375..859c1c187d0 100644
--- a/plugins/lombok/testData/compile/configSimple.kt
+++ b/plugins/lombok/testData/compile/configSimple.kt
@@ -15,13 +15,12 @@ public class GetterTest {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = GetterTest()
+fun box(): String {
+    val obj = GetterTest()
 
-        obj.primitiveBoolean
-        obj.getPrimitiveBoolean()
-    }
+    obj.primitiveBoolean
+    obj.getPrimitiveBoolean()
+    return "OK"
 }
 
 //FILE: lombok.config
diff --git a/plugins/lombok/testData/compile/data.kt b/plugins/lombok/testData/compile/data.kt
index d74f6d861be..45b66232eb8 100644
--- a/plugins/lombok/testData/compile/data.kt
+++ b/plugins/lombok/testData/compile/data.kt
@@ -27,18 +27,17 @@ import lombok.*;
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = DataExample("name")
-        obj.getName()
-        assertEquals(obj.name, "name")
-        obj.getTags()
-        val tags = obj.tags
-        obj.setScore(1.5)
-        assertEquals(obj.score, 1.5)
-        obj.score = 2.5
-        assertEquals(obj.score, 2.5)
+fun box(): String {
+    val obj = DataExample("name")
+    obj.getName()
+    assertEquals(obj.name, "name")
+    obj.getTags()
+    val tags = obj.tags
+    obj.setScore(1.5)
+    assertEquals(obj.score, 1.5)
+    obj.score = 2.5
+    assertEquals(obj.score, 2.5)
 
-        val ex: DataExample.Exercise<Int> = DataExample.Exercise.of("name", 12)
-    }
+    val ex: DataExample.Exercise<Int> = DataExample.Exercise.of("name", 12)
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/genericsAccessors.kt b/plugins/lombok/testData/compile/genericsAccessors.kt
index ca61daaef17..2091242bceb 100644
--- a/plugins/lombok/testData/compile/genericsAccessors.kt
+++ b/plugins/lombok/testData/compile/genericsAccessors.kt
@@ -26,13 +26,13 @@ public class GenericsTest<A, B> {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = GenericsTest<String, Boolean>()
-        val age: Int = obj.getAge();
-        obj.setFieldA("fooo");
-        val a: String = obj.getFieldA();
-        val b: Boolean? = obj.getFieldB();
-        obj.setFieldC(java.util.HashMap<String, Boolean>());
-    }
+fun box(): String {
+    val obj = GenericsTest<String, Boolean>()
+    val age: Int = obj.getAge();
+    obj.setFieldA("fooo");
+    val a: String = obj.getFieldA();
+    val b: Boolean? = obj.getFieldB();
+    obj.setFieldC(java.util.HashMap<String, Boolean>());
+    return "OK"
 }
+
diff --git a/plugins/lombok/testData/compile/genericsConstructors.kt b/plugins/lombok/testData/compile/genericsConstructors.kt
index 94bdf67716e..076de7f37bf 100644
--- a/plugins/lombok/testData/compile/genericsConstructors.kt
+++ b/plugins/lombok/testData/compile/genericsConstructors.kt
@@ -21,9 +21,8 @@ public class ConstructorExample<A, B> {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val generated = ConstructorExample<Long, Boolean>(12, 42L, true)
-        val generatedReq = ConstructorExample<String, Boolean>("234");
-    }
+fun box(): String {
+    val generated = ConstructorExample<Long, Boolean>(12, 42L, true)
+    val generatedReq = ConstructorExample<String, Boolean>("234");
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/genericsConstructorsStatic.kt b/plugins/lombok/testData/compile/genericsConstructorsStatic.kt
index 95b117c309b..dc7a2fb74d3 100644
--- a/plugins/lombok/testData/compile/genericsConstructorsStatic.kt
+++ b/plugins/lombok/testData/compile/genericsConstructorsStatic.kt
@@ -21,11 +21,10 @@ public class ConstructorExample<A, B> {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val generated: ConstructorExample<Long, Boolean> = ConstructorExample.of(12, 42L, true)
-        assertEquals(generated.name, 42L)
-        val generatedReq: ConstructorExample<String, Boolean> = ConstructorExample.of("234")
-        assertEquals(generatedReq.name, "234")
-    }
+fun box(): String {
+    val generated: ConstructorExample<Long, Boolean> = ConstructorExample.of(12, 42L, true)
+    assertEquals(generated.name, 42L)
+    val generatedReq: ConstructorExample<String, Boolean> = ConstructorExample.of("234")
+    assertEquals(generatedReq.name, "234")
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/getters.kt b/plugins/lombok/testData/compile/getters.kt
index 9f6807fc258..c61dd83ad39 100644
--- a/plugins/lombok/testData/compile/getters.kt
+++ b/plugins/lombok/testData/compile/getters.kt
@@ -25,30 +25,30 @@ public class GetterTest {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = GetterTest()
-        val getter = obj.getAge()
-        val property = obj.age
+fun box(): String {
+    val obj = GetterTest()
+    val getter = obj.getAge()
+    val property = obj.age
 
-        //todo kotlin doesn't see isBoolean methods as property
-//        obj.primitiveBoolean
-        obj.isPrimitiveBoolean()
+    //todo kotlin doesn't seee isBoolean methods as property
+//    obj.primitiveBoolean
+    obj.isPrimitiveBoolean()
 
-        obj.boxedBoolean
-        obj.getBoxedBoolean()
+    obj.boxedBoolean
+    obj.getBoxedBoolean()
 
-        //shouldn't be accesible from here
-//        obj.getName()
+    //shouldn't be accesible from here
+//    obj.getName()
 
-//        obj.getInvisible()
+//    obj.getInvisible()
 
-        OverridenGetterTest().usage()
-    }
+    OverridenGetterTest().usage()
+    return "OK"
+}
 
-    class OverridenGetterTest : GetterTest() {
-        fun usage() {
-            getName()
-        }
+class OverridenGetterTest : GetterTest() {
+    fun usage() {
+        getName()
     }
 }
+
diff --git a/plugins/lombok/testData/compile/gettersClassLevel.kt b/plugins/lombok/testData/compile/gettersClassLevel.kt
index 028733acc74..989279d1709 100644
--- a/plugins/lombok/testData/compile/gettersClassLevel.kt
+++ b/plugins/lombok/testData/compile/gettersClassLevel.kt
@@ -23,26 +23,25 @@ public class ClassLevelGetterTest {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = ClassLevelGetterTest()
-        val getter = obj.getAge()
-        val property = obj.age
+fun box(): String {
+    val obj = ClassLevelGetterTest()
+    val getter = obj.getAge()
+    val property = obj.age
 
-        obj.isPrimitiveBoolean()
+    obj.isPrimitiveBoolean()
 
-        obj.boxedBoolean
-        obj.getBoxedBoolean()
+    obj.boxedBoolean
+    obj.getBoxedBoolean()
 
-        //shouldn't be accesible from here
+    //shouldn't be accesible from here
 //        obj.getName()
 
-        OverridenGetterTest().usage()
-    }
+    OverridenGetterTest().usage()
+    return "OK"
+}
 
-    class OverridenGetterTest : ClassLevelGetterTest() {
-        fun usage() {
-            getName()
-        }
+class OverridenGetterTest : ClassLevelGetterTest() {
+    fun usage() {
+        getName()
     }
 }
diff --git a/plugins/lombok/testData/compile/gettersFluent.kt b/plugins/lombok/testData/compile/gettersFluent.kt
index 325fc39f19b..2a88ae55d10 100644
--- a/plugins/lombok/testData/compile/gettersFluent.kt
+++ b/plugins/lombok/testData/compile/gettersFluent.kt
@@ -27,24 +27,23 @@ public class FluentTest {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = FluentTest()
-        assertEquals(obj.age(), 10)
+fun box(): String {
+    val obj = FluentTest()
+    assertEquals(obj.age(), 10)
 
-        obj.primitiveBoolean()
+    obj.primitiveBoolean()
 
-        obj.boxedBoolean()
+    obj.boxedBoolean()
 
-        obj.overrideAnnotation
-        obj.getOverrideAnnotation()
+    obj.overrideAnnotation
+    obj.getOverrideAnnotation()
 
-        OverridenGetterTest().usage()
-    }
+    OverridenGetterTest().usage()
+    return "OK"
+}
 
-    class OverridenGetterTest : FluentTest() {
-        fun usage() {
-            name()
-        }
+class OverridenGetterTest : FluentTest() {
+    fun usage() {
+        name()
     }
 }
diff --git a/plugins/lombok/testData/compile/noArgsConstructor.kt b/plugins/lombok/testData/compile/noArgsConstructor.kt
index e815866aea5..541f038e723 100644
--- a/plugins/lombok/testData/compile/noArgsConstructor.kt
+++ b/plugins/lombok/testData/compile/noArgsConstructor.kt
@@ -22,9 +22,9 @@ public class ConstructorExample {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val existing = ConstructorExample("existing")
-        val generated = ConstructorExample()
-    }
+fun box(): String {
+    val existing = ConstructorExample("existing")
+    val generated = ConstructorExample()
+    return "OK"
 }
+
diff --git a/plugins/lombok/testData/compile/noArgsConstructorStatic.kt b/plugins/lombok/testData/compile/noArgsConstructorStatic.kt
index eda83b6a69f..5112b2714f6 100644
--- a/plugins/lombok/testData/compile/noArgsConstructorStatic.kt
+++ b/plugins/lombok/testData/compile/noArgsConstructorStatic.kt
@@ -22,9 +22,9 @@ public class ConstructorExample {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val existing: ConstructorExample = ConstructorExample("existing")
-        val generated: ConstructorExample = ConstructorExample.make()
-    }
+fun box(): String {
+    val existing: ConstructorExample = ConstructorExample("existing")
+    val generated: ConstructorExample = ConstructorExample.make()
+    return "OK"
 }
+
diff --git a/plugins/lombok/testData/compile/nullability.kt b/plugins/lombok/testData/compile/nullability.kt
index 9ae80d72579..a6bdb549694 100644
--- a/plugins/lombok/testData/compile/nullability.kt
+++ b/plugins/lombok/testData/compile/nullability.kt
@@ -14,10 +14,9 @@ public class GetterSetterExample {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = GetterSetterExample()
-        val age: Int = obj.getAge()
-        val name: String? = obj.getName()
-    }
+fun box(): String {
+    val obj = GetterSetterExample()
+    val age: Int = obj.getAge()
+    val name: String? = obj.getName()
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/processErrors/annotationTypes.kt b/plugins/lombok/testData/compile/processErrors/annotationTypes.kt
index 0061aca19ea..72858177e91 100644
--- a/plugins/lombok/testData/compile/processErrors/annotationTypes.kt
+++ b/plugins/lombok/testData/compile/processErrors/annotationTypes.kt
@@ -23,8 +23,7 @@ public class Foo {
 //FILE: test.kt
 
 @Anno
-class Test {
-    fun run() {
-        val obj = Foo()
-    }
+fun box(): String {
+    val obj = Foo()
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/propertyFromSuper.kt b/plugins/lombok/testData/compile/propertyFromSuper.kt
index 1731a594ebd..315ed48c52d 100644
--- a/plugins/lombok/testData/compile/propertyFromSuper.kt
+++ b/plugins/lombok/testData/compile/propertyFromSuper.kt
@@ -13,8 +13,7 @@ public class ChildClass extends ParentClass {}
 
 
 //FILE: test.kt
-class Test {
-    fun run() {
-        ChildClass::class
-    }
-}
\ No newline at end of file
+fun box(): String {
+    ChildClass::class
+    return "OK"
+}
diff --git a/plugins/lombok/testData/compile/requiredArgsConstructor.kt b/plugins/lombok/testData/compile/requiredArgsConstructor.kt
index 4de574249de..c625dfc51d9 100644
--- a/plugins/lombok/testData/compile/requiredArgsConstructor.kt
+++ b/plugins/lombok/testData/compile/requiredArgsConstructor.kt
@@ -33,9 +33,9 @@ public class ConstructorExample {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val generated = ConstructorExample("foo", true)
-        assertEquals(generated.foo, "foo")
-    }
+fun box(): String {
+    val generated = ConstructorExample("foo", true)
+    assertEquals(generated.foo, "foo")
+    return "OK"
 }
+
diff --git a/plugins/lombok/testData/compile/requiredArgsConstructorStatic.kt b/plugins/lombok/testData/compile/requiredArgsConstructorStatic.kt
index 4c37176c4ac..e47e90eec9d 100644
--- a/plugins/lombok/testData/compile/requiredArgsConstructorStatic.kt
+++ b/plugins/lombok/testData/compile/requiredArgsConstructorStatic.kt
@@ -35,9 +35,8 @@ public class ConstructorExample {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val generated = ConstructorExample.build("foo", true, 12)
-        assertEquals(generated.foo, "foo")
-    }
+fun box(): String {
+    val generated = ConstructorExample.build("foo", true, 12)
+    assertEquals(generated.foo, "foo")
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/setters.kt b/plugins/lombok/testData/compile/setters.kt
index acccd14a95f..122d4880a7c 100644
--- a/plugins/lombok/testData/compile/setters.kt
+++ b/plugins/lombok/testData/compile/setters.kt
@@ -20,25 +20,24 @@ public class SetterTest {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = SetterTest()
-        obj.setAge(42)
-        obj.age = 42
+fun box(): String {
+    val obj = SetterTest()
+    obj.setAge(42)
+    obj.age = 42
 
-        //synthetic property generated only when there is a getter
+    //synthetic property generated only when there is a getter
 //        obj.primitiveBoolean = false
-        obj.setPrimitiveBoolean(true)
+    obj.setPrimitiveBoolean(true)
 
-        //shouldn't be accesible from here
+    //shouldn't be accesible from here
 //        obj.setName("abc")
 
-        OverridenGetterTest().usage()
-    }
+    OverridenGetterTest().usage()
+    return "OK"
+}
 
-    class OverridenGetterTest : SetterTest() {
-        fun usage() {
-            setName("abc")
-        }
+class OverridenGetterTest : SetterTest() {
+    fun usage() {
+        setName("abc")
     }
 }
diff --git a/plugins/lombok/testData/compile/settersClassLevel.kt b/plugins/lombok/testData/compile/settersClassLevel.kt
index 824dcd01557..9ce24a62d5b 100644
--- a/plugins/lombok/testData/compile/settersClassLevel.kt
+++ b/plugins/lombok/testData/compile/settersClassLevel.kt
@@ -23,18 +23,17 @@ public class SetterTest {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = SetterTest()
-        obj.setAge(42)
-        assertEquals(obj.age, 42)
-        obj.age = 43
-        assertEquals(obj.age, 43)
+fun box(): String {
+    val obj = SetterTest()
+    obj.setAge(42)
+    assertEquals(obj.age, 42)
+    obj.age = 43
+    assertEquals(obj.age, 43)
 
-        obj.setPrimitiveBoolean(true)
+    obj.setPrimitiveBoolean(true)
 
-        
-//        no setters generated for final variable
-//        obj.setFinalName("error")
-    }
+//    no setters generated for final variable
+//    obj.setFinalName("error")
+    return "OK"
 }
+
diff --git a/plugins/lombok/testData/compile/settersVariations.kt b/plugins/lombok/testData/compile/settersVariations.kt
index 787f055f2a7..dae6d2d7ae6 100644
--- a/plugins/lombok/testData/compile/settersVariations.kt
+++ b/plugins/lombok/testData/compile/settersVariations.kt
@@ -25,15 +25,13 @@ public class SetterTest {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = SetterTest()
-        obj.fluent(12)
-        assertEquals(obj.fluent(), 12)
-        obj.setChained("zz").getChained()
-        assertEquals(obj.getChained(), "zz")
-        obj.whyNotBoth("zzz").whyNotBoth()
-        assertEquals(obj.whyNotBoth(), "zzz")
-    }
-
+fun box(): String {
+    val obj = SetterTest()
+    obj.fluent(12)
+    assertEquals(obj.fluent(), 12)
+    obj.setChained("zz").getChained()
+    assertEquals(obj.getChained(), "zz")
+    obj.whyNotBoth("zzz").whyNotBoth()
+    assertEquals(obj.whyNotBoth(), "zzz")
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/simple.kt b/plugins/lombok/testData/compile/simple.kt
index 3ac6abef25c..555fcaec85b 100644
--- a/plugins/lombok/testData/compile/simple.kt
+++ b/plugins/lombok/testData/compile/simple.kt
@@ -13,10 +13,9 @@ public class GetterSetterExample {
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = GetterSetterExample()
-        val getter = obj.getAge()
-        val property = obj.age
-    }
+fun box(): String {
+    val obj = GetterSetterExample()
+    val getter = obj.getAge()
+    val property = obj.age
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/value.kt b/plugins/lombok/testData/compile/value.kt
index 08df7bc2686..77856615b9b 100644
--- a/plugins/lombok/testData/compile/value.kt
+++ b/plugins/lombok/testData/compile/value.kt
@@ -28,17 +28,16 @@ import lombok.*;
 
 //FILE: test.kt
 
-class Test {
-    fun run() {
-        val obj = ValueExample("name", 12, 4.5)
-        assertEquals(obj.getName(), "name")
-        assertEquals(obj.name, "name")
-        assertEquals(obj.getAge(), 12)
-        assertEquals(obj.age, 12)
-        assertEquals(obj.score, 4.5)
+fun box(): String {
+    val obj = ValueExample("name", 12, 4.5)
+    assertEquals(obj.getName(), "name")
+    assertEquals(obj.name, "name")
+    assertEquals(obj.getAge(), 12)
+    assertEquals(obj.age, 12)
+    assertEquals(obj.score, 4.5)
 
-        val ex: ValueExample.Exercise<Int> = ValueExample.Exercise.of("nam1e", 42)
-        assertEquals(ex.name, "nam1e")
-        assertEquals(ex.value, 42)
-    }
+    val ex: ValueExample.Exercise<Int> = ValueExample.Exercise.of("nam1e", 42)
+    assertEquals(ex.name, "nam1e")
+    assertEquals(ex.value, 42)
+    return "OK"
 }
diff --git a/plugins/lombok/testData/compile/with.kt b/plugins/lombok/testData/compile/with.kt
index 4bcbe5fd970..34c751602e8 100644
--- a/plugins/lombok/testData/compile/with.kt
+++ b/plugins/lombok/testData/compile/with.kt
@@ -18,11 +18,9 @@ public class WithExample {
 
 //FILE: test.kt
 
-class Test {
-
-    fun run() {
-        val obj: WithExample = WithExample().withAge(16).withName("fooo")
-        assertEquals(obj.getName(), "fooo")
-        assertEquals(obj.age, 16)
-    }
+fun box(): String {
+    val obj: WithExample = WithExample().withAge(16).withName("fooo")
+    assertEquals(obj.getName(), "fooo")
+    assertEquals(obj.age, 16)
+    return "OK"
 }