Add a wait in order to avoid timeout