fix minor cleanup issue