Fix missing after callback bug