diff --git a/service/lib/agama/storage/config_size_solver.rb b/service/lib/agama/storage/config_size_solver.rb
index 1098c0d618..69449ae8ce 100644
--- a/service/lib/agama/storage/config_size_solver.rb
+++ b/service/lib/agama/storage/config_size_solver.rb
@@ -103,7 +103,7 @@ def size_from_product(config)
         config_builder.default_size(path, having_paths: paths, with_snapshots: snapshots)
       end
 
-      # @param config [Configs::Partition, Configs::LogicalVolume]
+      # @param device [Y2Storage::Device]
       # @return [Configs::Size]
       def size_from_device(device)
         Configs::Size.new.tap do |config|