Fixing dialog size issue