Fix import error when install with PIP