fix unit tests with import