Return to working code