refact imports in test case classed