fix(tests): Use updated annotation type