Fix OK Button on dialog box