Update to fix test