Fixed a bug with the dialog