Updated the atom decorator tests