rewrite the TestModels