Minor tweak to match Impy repo changes.