Fix small mistakes...