Fixed code mistake

Fixed code mistake