Fix containerWait check