diff --git a/lib/lattice/Lattice_trace.h b/lib/lattice/Lattice_trace.h
index f97c8d97..f0951cd4 100644
--- a/lib/lattice/Lattice_trace.h
+++ b/lib/lattice/Lattice_trace.h
@@ -42,9 +42,9 @@ inline auto trace(const Lattice<vobj> &lhs)
   -> Lattice<decltype(trace(lhs._odata[0]))>
 {
   Lattice<decltype(trace(lhs._odata[0]))> ret(lhs._grid);
-  parallel_for(int ss=0;ss<lhs._grid->oSites();ss++){
+  accelerator_loop( ss, lhs, {
     ret._odata[ss] = trace(lhs._odata[ss]);
-  }
+  });
   return ret;
 };
     
@@ -55,9 +55,9 @@ template<int Index,class vobj>
 inline auto TraceIndex(const Lattice<vobj> &lhs) -> Lattice<decltype(traceIndex<Index>(lhs._odata[0]))>
 {
   Lattice<decltype(traceIndex<Index>(lhs._odata[0]))> ret(lhs._grid);
-  parallel_for(int ss=0;ss<lhs._grid->oSites();ss++){
+  accelerator_loop( ss, lhs, {
     ret._odata[ss] = traceIndex<Index>(lhs._odata[ss]);
-  }
+  });
   return ret;
 };