Revert code was correct