Adopt import statements to reflect testing env