fix(tests): Use updated annotation type

Closes #948