Fix import in readme