Revert code to original version