ORD-KEF-FCO on Iceland Air - immigration where?

For those of you that have done the Iceland Air routes with a connection, do we do immigration at KEF or at final stop? Our route is ORD-KEF-FCO and vice versa. We have two hours on the way over, but on the way home only have 45 minutes.

Thanks :)