Tiny bug fixes