Couple small fixes.