Fixes nameerror on prod

Summary: https://app.getsentry.com/inbox/prod/group/38940736/ -- I mistakenly removed an import while refactoring this part of the code.

Test Plan: That's an import.

Reviewers: emfree

Reviewed By: emfree

Differential Revision: https://review.inboxapp.com/D680