getting import for tests right