Fix import error

A little leftover cruft, I fixed it like a boss.