Fixed Second broken code

Fixed Second broken code