actually fixed the dialog box