Forgot to re-add the on close dialog -- fixed