Undid if statement change