Fix the UI refresh callback, and by extension:  everything that relies on it