Update to revision handling