Add callback flag check