Import rescue fix

Super small changes.