Use `is_a?` instead of `class`