diff --git a/plugins/tolerance_time/db/migrate/20141119204010_add_timestamps_to_tolerance.rb b/plugins/tolerance_time/db/migrate/20141119204010_add_timestamps_to_tolerance.rb
new file mode 100644
index 0000000..34f09f4
--- /dev/null
+++ b/plugins/tolerance_time/db/migrate/20141119204010_add_timestamps_to_tolerance.rb
@@ -0,0 +1,5 @@
+class AddTimestampsToTolerance < ActiveRecord::Migration
+  def change
+    add_timestamps :tolerance_time_plugin_tolerances
+  end
+end
diff --git a/plugins/tolerance_time/test/unit/comment_test.rb b/plugins/tolerance_time/test/unit/comment_test.rb
index b124d83..959522e 100644
--- a/plugins/tolerance_time/test/unit/comment_test.rb
+++ b/plugins/tolerance_time/test/unit/comment_test.rb
@@ -3,8 +3,8 @@ require File.dirname(__FILE__) + '/../../../../test/test_helper'
 class CommentTest < ActiveSupport::TestCase
   should 'create a publication after posting a comment' do
     article = fast_create(Article, :profile_id => fast_create(Person).id)
-    comment = Comment.new(:author_id => fast_create(Person).id, :body => 'Hello There!', :source_id => article.id)
-    assert_difference ToleranceTimePlugin::Publication, :count do
+    comment = Comment.new(:author => fast_create(Person), :body => 'Hello There!', :source => article)
+    assert_difference 'ToleranceTimePlugin::Publication.count', 1 do
       comment.save!
     end
     assert_not_nil ToleranceTimePlugin::Publication.find_by_target(comment)
@@ -13,7 +13,7 @@ class CommentTest < ActiveSupport::TestCase
   should 'destroy publication if the comment is destroyed' do
     profile = fast_create(Profile)
     article = fast_create(Article, :profile_id => profile.id)
-    comment = fast_create(Comment, :source_id => article.id)
+    comment = fast_create(Comment, :source_id => article)
     comment_publication = ToleranceTimePlugin::Publication.create!(:target => comment)
     comment.destroy
     assert_raise ActiveRecord::RecordNotFound do
diff --git a/plugins/tolerance_time/test/unit/tolerance_time_plugin/publication_test.rb b/plugins/tolerance_time/test/unit/tolerance_time_plugin/publication_test.rb
index 199d0b4..648dd3e 100644
--- a/plugins/tolerance_time/test/unit/tolerance_time_plugin/publication_test.rb
+++ b/plugins/tolerance_time/test/unit/tolerance_time_plugin/publication_test.rb
@@ -4,13 +4,14 @@ class ToleranceTimePlugin::PublicationTest < ActiveSupport::TestCase
   should 'validate presence of target' do
     publication = ToleranceTimePlugin::Publication.new
     publication.valid?
-    assert publication.errors.invalid?(:target_id)
-    assert publication.errors.invalid?(:target_type)
+    assert publication.errors[:target_id].present?
+    assert publication.errors[:target_type].present?
 
     publication.target = fast_create(Article)
     publication.valid?
-    assert !publication.errors.invalid?(:target_id)
-    assert !publication.errors.invalid?(:target_type)
+
+    assert !publication.errors[:target_id].present?
+    assert !publication.errors[:target_type].present?
   end
 
   should 'validate uniqueness of target' do
@@ -19,7 +20,7 @@ class ToleranceTimePlugin::PublicationTest < ActiveSupport::TestCase
     p2 = ToleranceTimePlugin::Publication.new(:target => target)
     p2.valid?
 
-    assert p2.errors.invalid?(:target_id)
+    assert p2.errors[:target_id].present?
   end
 
   should 'be able to find publication by target' do
diff --git a/plugins/tolerance_time/test/unit/tolerance_time_plugin/tolerance_test.rb b/plugins/tolerance_time/test/unit/tolerance_time_plugin/tolerance_test.rb
index 44b8876..f2ffad6 100644
--- a/plugins/tolerance_time/test/unit/tolerance_time_plugin/tolerance_test.rb
+++ b/plugins/tolerance_time/test/unit/tolerance_time_plugin/tolerance_test.rb
@@ -4,11 +4,11 @@ class ToleranceTimePlugin::ToleranceTest < ActiveSupport::TestCase
   should 'validate presence of profile' do
     tolerance = ToleranceTimePlugin::Tolerance.new
     tolerance.valid?
-    assert tolerance.errors.invalid?(:profile_id)
+    assert tolerance.errors[:profile_id].present?
 
     tolerance.profile = fast_create(Profile)
     tolerance.valid?
-    assert !tolerance.errors.invalid?(:profile_id)
+    assert !tolerance.errors[:profile_id].present?
   end
 
   should 'validate uniqueness of profile' do
@@ -17,7 +17,7 @@ class ToleranceTimePlugin::ToleranceTest < ActiveSupport::TestCase
     t2 = ToleranceTimePlugin::Tolerance.new(:profile => profile)
     t2.valid?
 
-    assert t2.errors.invalid?(:profile_id)
+    assert t2.errors[:profile_id].present?
   end
 
   should 'validate integer format for comment and content tolerance' do
@@ -27,8 +27,8 @@ class ToleranceTimePlugin::ToleranceTest < ActiveSupport::TestCase
     tolerance.comment_tolerance = 'sdfa'
     tolerance.content_tolerance = 4.5
     tolerance.valid?
-    assert tolerance.errors.invalid?(:comment_tolerance)
-    assert tolerance.errors.invalid?(:content_tolerance)
+    assert tolerance.errors[:comment_tolerance].present?
+    assert tolerance.errors[:content_tolerance].present?
 
     tolerance.comment_tolerance = 3
     tolerance.content_tolerance = 6
--
libgit2 0.21.2