Put code back in place