1 files changed, 2 insertions, 3 deletions

diff --git a/test/unit/node_test.rb b/test/unit/node_test.rb
index 2558ca3..a7405f0 100644
--- a/test/unit/node_test.rb
+++ b/test/unit/node_test.rb
@@ -30,7 +30,7 @@ class NodeTest < ActiveSupport::TestCase
     assert_not_nil @first_child.draft
     assert_nil @first_child.draft.user
     @first_child.find_or_create_draft @user1
-    assert_equal @user1, @first_child.draft.user
+    assert_equal @user1, @first_child.user
   end
   
   def test_unique_path_returns_an_array
@@ -121,9 +121,8 @@ class NodeTest < ActiveSupport::TestCase
     node = Node.create :slug => "third_child"
     node.move_to_child_of @root
     node.publish_draft!
-
     node.find_or_create_draft @user1
-
+    assert_equal @user1, node.user
     assert_raise(RuntimeError) do
       node.find_or_create_draft @user2
     end