lib/orders/order_book.rb in orders-0.0.2 vs lib/orders/order_book.rb in orders-0.0.3

@@ -16,18 +16,20 @@
     end
 
     def add item
       if item.price > 0
         @changed = true # Marking DOM as changed
+        item.order_book = self
         super
       else
         self
       end
     end
 
     # Does not call super!
     def remove item
       @changed = true if delete index item # Marking DOM as changed
+      item.order_book = nil
       self
     end
   end
 end