Bring back needed code