Tiny bugs fix