fixing errors due to extension class changes