Fixing test issue in code.