// we register a custom UpdateDataCommandHandler that ensures to
// update the tree structure
this.bodyDataLayer.unregisterCommandHandler(UpdateDataCommand.class);
this.bodyDataLayer.registerCommandHandler(new UpdateDataCommandHandler(this.bodyDataLayer) {
    @SuppressWarnings("unchecked")
    @Override
    protected boolean doCommand(UpdateDataCommand command) {
        if (super.doCommand(command)) {
            T o = ((IRowDataProvider<T>) BodyLayerStack.this.bodyDataProvider).getRowObject(command.getRowPosition());
            int rowIndex = BodyLayerStack.this.sortedList.indexOf(o);
            if (rowIndex >= 0) {
                BodyLayerStack.this.sortedList.set(rowIndex, o);
            }
            return true;
        }
        return false;
    }
});