Better place to import unittest