Attempt to fix broken future_wait test