Oops forgot to add the on_close callback.