Add a wait to address issue on CI