Attempt to fix GUI bug