Quick UI Fix