diff --git a/src/main/java/com/sparrowwallet/drongo/wallet/WalletModel.java b/src/main/java/com/sparrowwallet/drongo/wallet/WalletModel.java
index 0517149..3ecfb83 100644
--- a/src/main/java/com/sparrowwallet/drongo/wallet/WalletModel.java
+++ b/src/main/java/com/sparrowwallet/drongo/wallet/WalletModel.java
@@ -7,6 +7,14 @@ public enum WalletModel {
         return valueOf(model.toUpperCase());
     }
 
+    public String getType() {
+        if(this == TREZOR_1 || this == TREZOR_T) {
+            return "trezor";
+        }
+
+        return this.toString();
+    }
+
     public String toDisplayString() {
         String line = this.toString().toLowerCase();
         String[] words = line.split("_");