Re-add lost class