fixed issue related to updated class file