Adding checks for unittests.