Fixed a broken reversion

Oops. Hopefully this works fine now