Changed QDialog to QWidget