Fix rollback in Command