Fix test warning