change method name
remove not needed get_class check