Minor code update.