Little bug fixes...