Extend the correct class this time