Fix test annotations