Add missing setDocument()