Minor code correction.