Removing wait_all as it's not really needed