Minor change in code.