     element = @getDialogForAttributeName(attributeName)
     element.classList.add("active")
 
-    input = element.querySelector("input[name='#{attributeName}']")
+    input = getInputForDialog(element, attributeName)
     input.value = @attributes[attributeName] ? ""
     input.select()
 

  getAttributeName = (element) ->
    element.getAttribute("data-attribute")

  getInputForDialog = (element) ->
    attributeName = getAttributeName(element)
    element.querySelector("input[name='#{attributeName}']")


  setAttribute: (dialogElement) ->
    attributeName = getAttributeName(dialogElement)
    value = getInputForDialog(dialogElement, attributeName).value
    @delegate?.toolbarDidUpdateAttribute(attributeName, value)

  removeAttribute: (dialogElement) ->
