Fix button close on dialog