Location of comment after decorator not handled correctly

source code:
```python

@dec # some comment
def a():
    return abcd
```

transformed:
```python
@dec
 # some comment
def a():
    return abcd
```