Fix tokenizing when docblocks are found