Set Class Name on Update, not just Import