Fixed two mistakes