Revert part of code