diff --git a/app/models/assistant.rb b/app/models/assistant.rb
index 5d630d3..80a7dad 100644
--- a/app/models/assistant.rb
+++ b/app/models/assistant.rb
@@ -1,6 +1,7 @@
 class Assistant < ApplicationRecord
   has_many :chats, dependent: :destroy
   belongs_to :user
+  belongs_to :library, optional: true
   enum status: { development: 0, ready: 1 }
   validates :name, presence: true
   validates :slack_channel_name, uniqueness: true, allow_blank: true
diff --git a/spec/factories/assistants.rb b/spec/factories/assistants.rb
index 62f4e01..46585c3 100644
--- a/spec/factories/assistants.rb
+++ b/spec/factories/assistants.rb
@@ -4,6 +4,7 @@
     llm_prompt { 'MyText' }
     libraries { '1' }
     input { 'some input' }
+    name { 'Test Assistant' }
     association :user
   end
 end