Some method updates for context roll back
Update the distribution