From bf452ecfe7c788dff378bc2b70a74e4a0ec52407 Mon Sep 17 00:00:00 2001
From: Neil Booth <kyuupichan@gmail.com>
Date: Sat, 2 Dec 2017 16:08:49 +0900
Subject: [PATCH] Make trezor code specific to trezor

Fixes #3308
---
 plugins/trezor/plugin.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/plugins/trezor/plugin.py b/plugins/trezor/plugin.py
index e5cc492c..57986f0d 100644
--- a/plugins/trezor/plugin.py
+++ b/plugins/trezor/plugin.py
@@ -183,11 +183,11 @@ class TrezorCompatiblePlugin(HW_PluginBase):
 
         if method == TIM_NEW:
             strength = 64 * (item + 2)  # 128, 192 or 256
-            u2f_counter = 0
-            skip_backup = False
-            client.reset_device(True, strength, passphrase_protection,
-                                pin_protection, label, language,
-                                u2f_counter, skip_backup)
+            args = [True, strength, passphrase_protection,
+                    pin_protection, label, language]
+            if self.device == 'TREZOR':
+                args.extend([0, False])  # u2f_counter, skip_backup
+            client.reset_device(*args)
         elif method == TIM_RECOVER:
             word_count = 6 * (item + 2)  # 12, 18 or 24
             client.step = 0