Add imports to make tests pass.