Fix for revert command