Fix UI notification issues with auth