Fix import test