you should do an instance of check instead of a get_class check IMO