Attempting to fix TestModel