Fix save loading (forgot to register the relevant classes).