Location of comment after decorator not handled correctly

[James4Ever0]

source code:

@dec # some comment
def a():
    return abcd

transformed:

@dec
 # some comment
def a():
    return abcd