Fixed small code mistake
Sorry, found another small mistake in my code