Fix import mudule error