Minor code corrections.