Fixed import statememt in test code.