remove 2 bugs

remove 2 bugs