Update reference to use custom annotations