Fixed the context menu dialog