diff --git a/app/models/piece_jointe.rb b/app/models/piece_jointe.rb
index d34bfc5ba..09bdedba1 100644
--- a/app/models/piece_jointe.rb
+++ b/app/models/piece_jointe.rb
@@ -1,8 +1,11 @@
 class PieceJointe < ActiveRecord::Base
   belongs_to :dossier
+  belongs_to :type_piece_jointe
 
   mount_uploader :content, PieceJointeUploader
 
+
+  # TODO: remove this function, find another way
   def self.get_array_id_pj_valid_for_dossier(dossier_id)
     @array_id_pj_valides = []
     where(dossier_id: dossier_id).each do |pj_valide|
diff --git a/spec/models/piece_jointe_spec.rb b/spec/models/piece_jointe_spec.rb
new file mode 100644
index 000000000..29c580cb5
--- /dev/null
+++ b/spec/models/piece_jointe_spec.rb
@@ -0,0 +1,12 @@
+require 'spec_helper'
+
+describe PieceJointe do
+  describe 'database columns' do
+    it { is_expected.to have_db_column(:content) }
+  end
+
+  describe 'associations' do
+    it { is_expected.to belong_to(:dossier) }
+    it { is_expected.to belong_to(:type_piece_jointe) }
+  end
+end
\ No newline at end of file