   @clearHighlightCanvas()
 
   for line, info of @markedLines
-    if @inTree info.model
+    if @inDisplay info.model
       path = @getHighlightPath info.model, info.style
       path.draw @highlightCtx
     else
       delete @markedLines[line]
 
   for id, info of @markedBlocks
-    if @inTree info.model
+    if @inDisplay info.model
       path = @getHighlightPath info.model, info.style
       path.draw @highlightCtx
     else
       delete @markedLines[id]
 
   for id, info of @extraMarks
-    if @inTree info.model
+    if @inDisplay info.model
       path = @getHighlightPath info.model, info.style
       path.draw @highlightCtx
     else


  # If there is an block that is being dragged,
  # draw it in gray
  if @draggingBlock? and @inTree @draggingBlock
    @view.getViewNodeFor(@draggingBlock).draw @highlightCtx, new @draw.Rectangle(
      @scrollOffsets.main.x,
      @scrollOffsets.main.y,

class FloatingBlockRecord
  constructor: (@block, @position) ->

Editor::inTree = (block) -> (block.container ? block).getDocument() is @tree
Editor::inDisplay = (block) -> (block.container ? block).getDocument() in @getDocuments()

# We can create floating blocks by dropping
# blocks without a highlight.
