spec/update_spec.rb in dm-rest-adapter-0.9.8 vs spec/update_spec.rb in dm-rest-adapter-0.9.9

@@ -18,10 +18,10 @@
         @book = Book.new(:id => 42,
                          :title => 'Starship Troopers',
                          :author => 'Robert Heinlein',
                          :created_at => DateTime.parse('2008-06-08T17:02:28Z'))
         @book.instance_eval { @new_record = false }
-        @repository.identity_map(Book).set(@book.key, @book)
+        @repository.identity_map(Book)[@book.key] = @book
         @book.title = "Mary Had a Little Lamb"
       end
     end
 
     it 'should do an HTTP PUT' do