Fixed an error on the LegacyModel