couple bug fixes