Fixed two incorrect lines causing errors