Set code back to working state