Add some annotation to let LKY understand