Extend correct class