couple small fixes