Fix import compile error