Minor code adjustment.