[models] Fix build break for unit tests.

Closes #450.