diff --git a/lib/shale/mapper.rb b/lib/shale/mapper.rb
index 4e8e208..59eb81d 100644
--- a/lib/shale/mapper.rb
+++ b/lib/shale/mapper.rb
@@ -144,7 +144,7 @@ def model(klass = nil)
       # Define attribute on class
       #
       # @param [Symbol] name Name of the attribute
-      # @param [Shale::Type::Value] type Type of the attribute
+      # @param [Class<Shale::Type::Value>] type Type of the attribute
       # @param [Boolean] collection Is the attribute a collection
       # @param [Proc] default Default value for the attribute
       #