Fix bugs #2