add this argument to after_click callback