Restore missing classes