Fix tiny bugs