Couple bug fixes