fix tests again :tired_face: