Aligned test to code update.