fix tests imports