Как мне исправить vim, чтобы правильно отступать сгибы, содержащие строки комментариев Python?

Я установил метод свертывания vim на отступ, который очень хорошо работает при написании Python, за исключением случаев, когда у меня есть строка комментария. Например, если у меня есть этот фрагмент кода:

def myFunction():
    # here is my comment
    myString = "hello"
    myInt = 2

Если я наведу курсор на строку комментария и наберу «za», я получаю сообщение об ошибке «E490: складка не найдена». Если у меня есть курсор на строке, начинающейся с "myString =", я получу свертку следующим образом:

def myFunction():
    # here is my comment
+--- 2 lines: myString = "hello" -------------------------

В обоих случаях я хотел бы получить эту свертку:

def myFunction():
+--- 3 lines: # here is my comment -------------------------

В основном, строка комментария должна обрабатываться как все остальное. Я не нашел ответа, поискав в Интернете. Любые идеи? Спасибо!