fixed some imports that were missing.

that's what you get for coding blind.