Python force move bug

[selfint]

Input:

from dataclasses import dataclass

@dataclass
class A:
    def __init__(self) -> None:
        ...

    def foo(self) -> None:
        ...

def main():
    ...

if __name__ == "__main__":
    main()

Trying to force move foo to be below @dataclass\nclass A fails dramatically.

This is probably because the order we are changing the text is wrong?