Fixing highlighting issue with @debug & @warn