fixing import in unittests
fixing tests