Issue 973441 2

As you have a `_updateUI` function, we could simply set it as the callback function.