add imports to BaseTestHelper.java