bug fixes...