fix bugs #2