[Main] right import package in testing files