Annotate statements in RestrictedFunctionsUnitTest