Update tests for annotated transformation