Add import logging in unittests