fix: pragma directives not picked up on return statement (#540)

fix: pragma directives not picked up on return statement (#540) #1547