Fix atomic unit tests