1 files changed, 7 insertions, 1 deletions
diff --git a/src/storage/root.zig b/src/storage/root.zig
index c121ca9..381dc45 100644
--- a/src/storage/root.zig
+++ b/src/storage/root.zig
@@ -3,7 +3,10 @@ const config = @import("../config.zig");
 const prompt = @import("../prompt.zig");
 
 pub const User = @import("user.zig");
-pub const Image = @import("image.zig");
+
+pub const ImageManager = @import("image-manager/root.zig");
+pub const Image = ImageManager.Image;
+
 pub const SessionManager = @import("session-manager/root.zig");
 pub const Session = SessionManager.Session;
 
@@ -11,6 +14,7 @@ const Self = @This();
 
 dir: std.fs.Dir,
 sessions: SessionManager = .empty,
+images: ImageManager = .empty,
 allocator: std.mem.Allocator,
 
 pub fn init(allocator: std.mem.Allocator) !Self {
@@ -56,6 +60,8 @@ pub fn init(allocator: std.mem.Allocator) !Self {
         else => return err,
     };
 
+    try self.images.init(&self);
+
     return self;
 }