fixed a couple of import statements