Highlighting does not work correctly within a script tag

For example:

`{% block head %}
    {{ parent() }}
    <script>
        $(function() {
            {% if country is not empty %}
                document.getElementById("country").value = "{{ country }}";
            {% endif %}
        });
    </script>
{% endblock %}
`
