BUG: fix update_all tests.