small model change
Revert "small model change"

This reverts commit 06fc77a22b9bd18009a13fc02ca29d3cb5dd0b62.