Prettied up the import statements on tests