Fixing a bug with bad call to updateUI