diff --git a/src/JustRelax_CPU.jl b/src/JustRelax_CPU.jl
index cb4134ee..5022bdb5 100644
--- a/src/JustRelax_CPU.jl
+++ b/src/JustRelax_CPU.jl
@@ -21,7 +21,7 @@ import JustRelax:
 
 import JustPIC._2D: numphases, nphases
 
-@init_parallel_stencil(Threads, Float64, 2)
+__init__() = @init_parallel_stencil(Threads, Float64, 2)
 
 include("common.jl")
 include("stokes/Stokes2D.jl")
@@ -52,7 +52,7 @@ import JustRelax:
 
 import JustPIC._3D: numphases, nphases
 
-@init_parallel_stencil(Threads, Float64, 3)
+__init__() = @init_parallel_stencil(Threads, Float64, 3)
 
 include("common.jl")
 include("stokes/Stokes3D.jl")
diff --git a/src/ext/AMDGPU/2D.jl b/src/ext/AMDGPU/2D.jl
index 712e3b17..d93070f4 100644
--- a/src/ext/AMDGPU/2D.jl
+++ b/src/ext/AMDGPU/2D.jl
@@ -31,7 +31,7 @@ import JustRelax:
 
 import JustPIC._2D: nphases, numphases
 
-@init_parallel_stencil(AMDGPU, Float64, 2)
+__init__() = @init_parallel_stencil(AMDGPU, Float64, 2)
 
 include("../../common.jl")
 include("../../stokes/Stokes2D.jl")
diff --git a/src/ext/AMDGPU/3D.jl b/src/ext/AMDGPU/3D.jl
index 6a566f15..7b00514e 100644
--- a/src/ext/AMDGPU/3D.jl
+++ b/src/ext/AMDGPU/3D.jl
@@ -31,7 +31,7 @@ import JustRelax:
 
 import JustPIC._3D: numphases, nphases
 
-@init_parallel_stencil(AMDGPU, Float64, 3)
+__init__() = @init_parallel_stencil(AMDGPU, Float64, 3)
 
 include("../../common.jl")
 include("../../stokes/Stokes3D.jl")
diff --git a/src/ext/CUDA/2D.jl b/src/ext/CUDA/2D.jl
index 0b83f7e6..6e871169 100644
--- a/src/ext/CUDA/2D.jl
+++ b/src/ext/CUDA/2D.jl
@@ -24,7 +24,7 @@ import JustRelax:
 
 import JustPIC._2D: numphases, nphases
 
-@init_parallel_stencil(CUDA, Float64, 2)
+__init__() = @init_parallel_stencil(CUDA, Float64, 2)
 
 include("../../common.jl")
 include("../../stokes/Stokes2D.jl")
diff --git a/src/ext/CUDA/3D.jl b/src/ext/CUDA/3D.jl
index c4725450..ee4a0007 100644
--- a/src/ext/CUDA/3D.jl
+++ b/src/ext/CUDA/3D.jl
@@ -23,7 +23,7 @@ import JustRelax:
 
 import JustPIC._3D: numphases, nphases
 
-@init_parallel_stencil(CUDA, Float64, 3)
+__init__() = @init_parallel_stencil(CUDA, Float64, 3)
 
 include("../../common.jl")
 include("../../stokes/Stokes3D.jl")