Get tests to work using example unittest.