Allow wait_for in multi_put