Make the unit test run properly - fix import