Fix i element bug