Couple bug fixes.