1 files changed, 3 insertions, 2 deletions
diff --git a/device.cc b/device.cc
index 0427d29..4f11e23 100644
--- a/device.cc
+++ b/device.cc
@@ -45,8 +45,6 @@ int Device::send(const void *packet) {
 }
 
 int Device::recv(void *packet) {
-  // FIXME change CommError return value to int (return value of
-  // hid_get_feature_report)
   int status;
   int retry_count = 0;
 
@@ -86,10 +84,13 @@ int Device::recv(void *packet) {
 Stick10::Stick10() {
   m_vid = 0x20a0;
   m_pid = 0x4108;
+  m_model = DeviceModel::PRO;
 }
 
 Stick20::Stick20() {
   m_vid = 0x20a0;
   m_pid = 0x4109;
   m_retry_timeout = std::chrono::milliseconds(500);
+  m_model = DeviceModel::STORAGE;
+
 }