Remove TestModel class