found solution for including class in test and try to fix problem with missing function from other module