fix tests (tm)