Fixing these two tests to use unittest.