Return code back to where it was