My mistake, reverted to original get_class()