Fixed deleted code