Fix loading of extension class