fix class inheritence