How can I check a damage source?

Actually figured it out! I just need to do the comparison using class!