Fixed deleted code by mistake

Please review