Fix import in test case