Fix the atomic ops test