Query bug fix, update UI