Minor change of code