1 files changed, 6 insertions, 2 deletions
diff --git a/unittest/test_multiple_devices.cc b/unittest/test_multiple_devices.cc
index f9e9ad2..183af4f 100644
--- a/unittest/test_multiple_devices.cc
+++ b/unittest/test_multiple_devices.cc
@@ -35,9 +35,11 @@ using namespace nitrokey;
 
 TEST_CASE("List devices", "[BASIC]") {
     shared_ptr<Stick20> d = make_shared<Stick20>();
-    auto v = d->enumerate();
+    auto v = Device::enumerate();
     REQUIRE(v.size() > 0);
     for (auto i : v){
+        if (i.m_deviceModel != DeviceModel::STORAGE)
+            continue;
         auto a = i.m_path;
         std::cout << a;
         d->set_path(a);
@@ -54,11 +56,13 @@ TEST_CASE("List devices", "[BASIC]") {
 
 TEST_CASE("Regenerate AES keys", "[BASIC]") {
     shared_ptr<Stick20> d = make_shared<Stick20>();
-    auto v = d->enumerate();
+    auto v = Device::enumerate();
     REQUIRE(v.size() > 0);
 
     std::vector<shared_ptr<Stick20>> devices;
     for (auto i : v){
+        if (i.m_deviceModel != DeviceModel::STORAGE)
+            continue;
         auto a = i.m_path;
         std::cout << a << endl;
         d = make_shared<Stick20>();