Fix small mistakes