minor: fix tests