fix issue where not all models were serialized