The change in logic looks fine.