2 bugs fix