Update tests due to previous change.