Fix
Attempted to fix UI dissapearing; is a java bug not me!