Why not put this into TestModel