diff --git a/plugins/atomicfu/atomicfu-compiler/src/org/jetbrains/kotlinx/atomicfu/compiler/backend/common/AbstractAtomicfuTransformer.kt b/plugins/atomicfu/atomicfu-compiler/src/org/jetbrains/kotlinx/atomicfu/compiler/backend/common/AbstractAtomicfuTransformer.kt
index 8d05dcb853faf..a4ff36c416981 100644
--- a/plugins/atomicfu/atomicfu-compiler/src/org/jetbrains/kotlinx/atomicfu/compiler/backend/common/AbstractAtomicfuTransformer.kt
+++ b/plugins/atomicfu/atomicfu-compiler/src/org/jetbrains/kotlinx/atomicfu/compiler/backend/common/AbstractAtomicfuTransformer.kt
@@ -110,7 +110,7 @@ abstract class AbstractAtomicfuTransformer(
         }
     }
 
-    abstract inner class AtomicPropertiesTransformer : IrElementTransformer<IrFunction?> {
+    abstract inner class AtomicPropertiesTransformer : IrTransformer<IrFunction?>() {
 
         override fun visitClass(declaration: IrClass, data: IrFunction?): IrStatement {
             val declarationsToBeRemoved = mutableListOf<IrDeclaration>()
@@ -387,7 +387,7 @@ abstract class AbstractAtomicfuTransformer(
             }
     }
 
-    abstract inner class AtomicFunctionCallTransformer : IrElementTransformer<IrFunction?> {
+    abstract inner class AtomicFunctionCallTransformer : IrTransformer<IrFunction?>() {
 
         override fun visitFunction(declaration: IrFunction, data: IrFunction?): IrStatement {
             return super.visitFunction(declaration, declaration)
@@ -705,7 +705,7 @@ abstract class AbstractAtomicfuTransformer(
      *
      * It's launched as a separate transformation stage to avoid recursive visiting.
      */
-    private inner class RemapValueParameters : IrElementTransformer<IrFunction?> {
+    private inner class RemapValueParameters : IrTransformer<IrFunction?>() {
 
         override fun visitFunction(declaration: IrFunction, data: IrFunction?): IrStatement {
             return super.visitFunction(declaration, declaration)
@@ -748,7 +748,7 @@ abstract class AbstractAtomicfuTransformer(
         }
     }
 
-    private inner class FinalTransformationChecker : IrElementTransformer<IrFunction?> {
+    private inner class FinalTransformationChecker : IrTransformer<IrFunction?>() {
         override fun visitFunction(declaration: IrFunction, data: IrFunction?): IrStatement {
             return super.visitFunction(declaration, declaration)
         }
diff --git a/plugins/atomicfu/atomicfu-compiler/src/org/jetbrains/kotlinx/atomicfu/compiler/backend/js/AtomicfuJsIrTransformer.kt b/plugins/atomicfu/atomicfu-compiler/src/org/jetbrains/kotlinx/atomicfu/compiler/backend/js/AtomicfuJsIrTransformer.kt
index 652f4e8f4cf94..490ca9233b590 100644
--- a/plugins/atomicfu/atomicfu-compiler/src/org/jetbrains/kotlinx/atomicfu/compiler/backend/js/AtomicfuJsIrTransformer.kt
+++ b/plugins/atomicfu/atomicfu-compiler/src/org/jetbrains/kotlinx/atomicfu/compiler/backend/js/AtomicfuJsIrTransformer.kt
@@ -17,7 +17,7 @@ import org.jetbrains.kotlin.ir.types.*
 import org.jetbrains.kotlin.ir.util.*
 import org.jetbrains.kotlin.ir.visitors.IrElementTransformerVoid
 import org.jetbrains.kotlin.ir.expressions.IrTypeOperator.*
-import org.jetbrains.kotlin.ir.visitors.IrElementTransformer
+import org.jetbrains.kotlin.ir.visitors.IrTransformer
 import org.jetbrains.kotlin.platform.isJs
 
 private const val AFU_PKG = "kotlinx.atomicfu"
@@ -95,7 +95,7 @@ class AtomicfuJsIrTransformer(private val context: IrPluginContext) {
         }
     }
 
-    private inner class AtomicTransformer : IrElementTransformer<IrFunction?> {
+    private inner class AtomicTransformer : IrTransformer<IrFunction?>() {
 
         override fun visitProperty(declaration: IrProperty, data: IrFunction?): IrStatement {
             // Support transformation for delegated properties: