diff --git a/pykokkos/core/visitors/workunit_visitor.py b/pykokkos/core/visitors/workunit_visitor.py
index ed82b0eb..807eafe4 100644
--- a/pykokkos/core/visitors/workunit_visitor.py
+++ b/pykokkos/core/visitors/workunit_visitor.py
@@ -295,7 +295,7 @@ def visit_Call(self, node: ast.Call) -> cppast.CallExpr:
             else:
                 return cppast.CallExpr(function, [args[0], f"pk_id_{work_unit}"])
 
-        atomic_fetch_op: re.Pattern = re.compile("atomic_*")
+        atomic_fetch_op: re.Pattern = re.compile("atomic_fetch_*")
         is_atomic_fetch_op: bool = atomic_fetch_op.match(name)
         is_atomic_increment: bool = name == "atomic_increment"
         is_atomic_compare_exchange: bool = name == "atomic_compare_exchange"