Add another wait to address CI failures