diff --git a/lib/vite_ruby/sri_extensions.rb b/lib/vite_ruby/sri_extensions.rb
index 0055060c45..7454d037bb 100644
--- a/lib/vite_ruby/sri_extensions.rb
+++ b/lib/vite_ruby/sri_extensions.rb
@@ -47,7 +47,7 @@ module ViteRuby::ManifestIntegrityExtension
       }
     end
 
-    imports = dev_server_running? ? [] : entries.flat_map { |entry| entry['imports'] }.compact
+    imports = dev_server_running? ? [] : entries.flat_map { |entry| import_chunks_for(entry) }
 
     {
       scripts: script_paths,