quick UI fix