fix minor problem