fix bug

修复bug#2 #3