Changed mock use of mock function to annotation.