Integrate with Test model