Update testing case for Atom when disabling rematerialization in