Minor change in code style