Fixing an issue in the code.