Small code corrections.