resolve problem with referencing old class