1 files changed, 2 insertions, 1 deletions
diff --git a/src/device.rs b/src/device.rs
index 5dc6724..9c6608d 100644
--- a/src/device.rs
+++ b/src/device.rs
@@ -233,6 +233,7 @@ pub trait Device: Authenticate + GetPasswordSafe + GenerateOtp {
     /// # Example
     ///
     /// ```no_run
+    /// use nitrokey::Device;
     /// # use nitrokey::CommandError;
     ///
     /// # fn try_main() -> Result<(), CommandError> {
@@ -787,7 +788,7 @@ impl Drop for Storage {
 
 impl Device for Storage {
     fn get_model(&self) -> Model {
-        Model::Pro
+        Model::Storage
     }
 }