Fix import exception