Is it correct to change your wait here