Fix UI feedback