Instead of using context.selectedIDs(), define a check.entity setter function