[minor] fix tests