Fixing accidentally deletion of needed code