- fix tests