fix extension class