Fix up the import path after #3332.

Fix up the import path after #3332.

@devoncarew @ussuri