Adding MC logic in budget enforcement