diff --git a/numba_dpex/core/compiler.py b/numba_dpex/core/compiler.py
index fc0074534b..cfb0e7d8ab 100644
--- a/numba_dpex/core/compiler.py
+++ b/numba_dpex/core/compiler.py
@@ -85,7 +85,7 @@ def compile_with_dpex(
             return_type=return_type,
             flags=flags,
             locals={},
-            pipeline_class=OffloadCompiler,
+            pipeline_class=KernelCompiler,
         )
     else:
         raise UnreachableError()