Extension of isomorphisms in abelian $p$-groups

It is proved that every isomorphism between any two subgroups of a group $G$ retaining the height of the elements in $G$ is extended to an automorphism of the group itself in the class of abelian $p$-groups without elements of infinite height if and only if $G$ is a closed group with finite Ulam invariants.