little bug fixes