fix: Check for class type exactly instead of with contains (#310)

fix: Check for class type exactly instead of with contains (#310) #902