Fix for hiding UI