Change if statement to check for class_name instead of name (#287)

Change if statement to check for class_name instead of name (#287) #3065