Works with node reference checkbox / radio buttons widgets

Looks node reference field has new widget available. Now we can use checkbox / radio buttons widgets for choose node reference value. Since Multi-column checkboxes radios already did a good formatting work for the widget. I thought it also can serve to node reference.
Here's one line change only. At 13rd line of module.

from
if ($widget['type'] == 'optionwidgets_buttons' || $widget['widget_type'] == 'optionwidgets_buttons') {
to
if ($widget['type'] == 'optionwidgets_buttons' || $widget['widget_type'] == 'optionwidgets_buttons' || $widget['widget_type'] == 'nodereference_buttons' || $widget['type'] == 'nodereference_buttons') {

thanks the module~