Fix context menu data context issue