Fix context menu