fixing a couple silly import errors