Another fix for future_wait test