Правильно вы же элементу $custom подставляете значение

if ($select.val() == customValue) {
            $element = $custom;
        }

а нужно $select