Add override for callback