fix tests wrt model passing