Update to fix failed test