s390/sclp: improve special wait psw logic