Add correct atomic blocks to benchmark