fix tests to import either