Fixes error in atomic tests