[S390] cio: improve unit check handling for internal operations