2 small fixes